Submission #2406672

---

Source Code Expand

#include <iostream>
#include <string>
#include <vector>
#include <stack>
using namespace std;

const long long mod = 998244353;

struct State {
	long long ty, a, b;
};
long long parse(string A) {
	long long J = 0;
	for (int i = 0; i < A.size(); i++) {
		J *= 10; J += (A[i] - '0'); J %= mod;
	}
	return J;
}
string S, T; vector<string>V1; vector<long long>U;
stack<State>I;

int main() {
	cin >> S; string V = "";
	for (int i = 0; i < S.size(); i++) {
		if (S[i] >= '0' && S[i] <= '9') V += S[i];
		else { if (V.size() >= 1) { T += "1"; V1.push_back(V); V = ""; } T += S[i]; }
	}
	if (V.size() >= 1) { T += "1"; V1.push_back(V); V = ""; }
	for (int i = 0; i < V1.size(); i++) U.push_back(parse(V1[i]));

	int cnt = 0;

	for (int i = 0; i < T.size(); i++) {
		if (T[i] == '(') {
			I.push(State{ 1,0,0 });
		}
		if (T[i] == '1') {
			I.push(State{ 2,U[cnt],0 });
			cnt++;
		}
		if (T[i] == '*') {
			I.push(State{ 3,0,0 });
		}
		if (T[i] == ',') {
			I.push(State{ 4,0,0 });
		}
		if (T[i] == ')') {
			State V3 = I.top(); I.pop();
			State V2 = I.top(); I.pop();
			State V1 = I.top(); I.pop();
			I.pop();
			if (V2.ty == 3) {
				if (V1.ty == 2 && V3.ty == 2) I.push(State{ 2,V1.a*V3.a%mod,0 });
				if (V1.ty == 2 && V3.ty == 5) I.push(State{ 5,V1.a*V3.a%mod,V1.a*V3.b%mod });
				if (V1.ty == 5 && V3.ty == 2) I.push(State{ 5,V1.a*V3.a%mod,V1.b*V3.a%mod });
				if (V1.ty == 5 && V3.ty == 5) I.push(State{ 2,(V1.a*V3.a + V1.b*V3.b) % mod });
			}
			if (V2.ty == 4) {
				I.push(State{ 5,V1.a,V3.a });
			}
		}
	}
	cout << I.top().a << endl;
	return 0;
}

Submission Info

Judge Result