자이의 프로그래밍
맞춰봐 본문
이상한 지문의 문제... 재귀적으로 모든 값을 체크하면서 수를 1~10까지밖에 모르므로 아래의 방식을 사용했다.
#include <iostream>
#include <string>
using namespace std;
int n;
int sign[10][10];
int ans[10];
bool check(int index) {
int sum = 0;
for (int i=index; i>=0; i--) {
sum += ans[i];
if (sign[i][index] == 0) {
if (sum != 0) return false;
} else if (sign[i][index] < 0) {
if (sum >= 0) return false;
} else if (sign[i][index] > 0) {
if (sum <= 0) return false;
}
}
return true;
}
bool go(int index) {
if (index == n) {
return true;
}
if (sign[index][index] == 0) {
ans[index] = 0;
return check(index) && go(index+1);
}
for (int i=1; i<=10; i++) {
ans[index] = sign[index][index]*i;
if (check(index) && go(index+1)) return true;
}
return false;
}
int main() {
cin >> n;
string s;
cin >> s;
int cnt = 0;
for (int i=0; i<n; i++) {
for (int j=i; j<n; j++) {
if (s[cnt] == '0') {
sign[i][j] = 0;
} else if (s[cnt] == '+') {
sign[i][j] = 1;
} else {
sign[i][j] = -1;
}
cnt += 1;
}
}
go(0);
for (int i=0; i<n; i++) {
cout << ans[i] << ' ';
}
cout << '\n';
return 0;
}