Xi_kor 2021. 7. 10. 09:53

이상한 지문의 문제... 재귀적으로 모든 값을 체크하면서 수를 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;
}