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; }