mayoko’s diary

プロコンとかいろいろ。

yukicoder No.265 数学のテスト

疲れた…
やっぱ構文解析難しいです。

解法

構文解析する。

ノリは↓と同じです。mayokoex.hatenablog.com

int d;
typedef string::const_iterator State;

vector<ll> expression(State& begin);
vector<ll> term(State& begin);

void print(const vector<ll>& x) {
    for (int i = 0; i <= d; i++) {
        cout << x[i];
        if (i != d) cout << " ";
    }
    cout << endl;
}

vector<ll> expression(State& begin) {
    vector<ll> ret(d+1);
    for (;;) {
        if (*begin == 'd') {
            begin++;
            begin++;
            auto x = expression(begin);
            for (int i = 1; i <= d; i++) ret[i-1] += x[i]*i;
            begin++;
        } else if (*begin == 'x' || isdigit(*begin)) {
            auto x = term(begin);
            for (int i = 0; i <= d; i++) ret[i] += x[i];
        } else if (*begin == '+') {
            begin++;
            auto x = expression(begin);
            for (int i = 0; i <= d; i++) ret[i] += x[i];
        }
        else break;
    }
    return ret;
}

vector<ll> term(State& begin) {
    ll c = 1;
    int degree = 0;
    for (;;) {
        if (*begin == 'x') degree++;
        else if (*begin == '*') {}
        else if (isdigit(*begin)) c *= (*begin-'0');
        else break;
        begin++;
    }
    vector<ll> ret(d+1);
    ret[degree] += c;
    return ret;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int N;
    cin >> N;
    cin >> d;
    string s;
    cin >> s;
    State begin = s.begin();
    auto x = expression(begin);
    print(x);
    return 0;
}