// subtask_7.txt, subtask_10.txt で, WAとなったので, ロジック修正. // -> if(befX == '#') ans = c; を 追加. // -> (1-2)+(3-4) = -2 を計算できるようになった. #include using namespace std; int main(){ // 1. 入力情報取得. string S; cin >> S; // 2. 括弧ごとに計算. int ans = 0; vector v; bool isBracket = false; char curX, befX = '#'; for(int i = 0; i < S.size(); i++){ curX = S[i]; // 2-1. 左括弧来たら, フラグ を true に 変更. if(curX == '('){ isBracket = true; continue; } // 2-2. 右括弧来たら, フラグ を false に 変更. if(curX == ')'){ isBracket = false; int c = 0; char bef = '#'; for(auto &cur : v){ if(cur == '+' || cur == '-') bef = cur; if(cur != '+' && cur != '-'){ if(bef == '#') c = (cur - '0'); if(bef == '+') c += (cur - '0'); if(bef == '-') c -= (cur - '0'); } // cout << "cur=" << cur << " c=" << c << endl; } if(befX == '#') ans = c; // subtask_7.txt, subtask_10.txt での WA対策. if(befX == '+') ans += c; if(befX == '-') ans -= c; // cout << "isBracket=" << isBracket << " befX=" << befX << " curX=" << curX << " c=" << c << " ans=" << ans << endl; // v を 初期化. v = vector(); continue; } // 2-3. 括弧以外は, 保存 or 計算. if(isBracket) v.push_back(curX); if(!isBracket && curX != '+' && curX != '-'){ if(befX == '#') ans = (curX - '0'); if(befX == '+') ans += (curX - '0'); if(befX == '-') ans -= (curX - '0'); // cout << "isBracket=" << isBracket << " befX=" << befX << " curX=" << curX << " ans=" << ans << endl; } // 2-4. 前回 の '+' or '-' を 保存. if((curX == '+' || curX == '-') && !isBracket) befX = curX; } // 3. 出力 ~ 後処理. // ex. // 0-(1-2)+(3+4)-5+6-(7+8+9)+0+1+2+3+4+5 = 0 // 5-(3-2+4)+(3-7-4+1)-5+6-0-(7+2+9)+5+0+1-2+3-4+(5+9)-3+7 = -3 // のはず. // // (1-2)+(3-4) = -1 が出力されたことに注意(正解は, -2 の はず). // -> subtask_7.txt, subtask_10.txt での WA の 原因と推測されるもの. cout << ans << endl; return 0; }