結果

問題 No.1677 mæx
ユーザー 夜
提出日時 2021-12-15 21:18:41
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 20 ms / 2,000 ms
コード長 2,237 bytes
コンパイル時間 823 ms
コンパイル使用メモリ 94,172 KB
実行使用メモリ 9,460 KB
最終ジャッジ日時 2024-07-23 20:01:37
合計ジャッジ時間 1,941 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 't expression(int&)':
main.cpp:98:1: warning: control reaches end of non-void function [-Wreturn-type]
   98 | }
      | ^

ソースコード

diff # 

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <cmath>
#include <stdio.h>
#include <queue>
#include <deque>
#include <cstdio>
#include <set>
#include <map>
#include <bitset>
#include <stack>
#include <cctype>
using namespace std;
string s;
int k, i = 0;
int e[3][3] = { {1,2,1},{2,0,0},{1,0,0} };
int a[3][3] = { {0,1,2},{1,1,2},{2,2,2} };
long long mod = 998244353;
typedef struct { long long x, y, z; }t;
t expression(int &i){
	if (s[i] == 'm') {
		i++;
		if (s[i] == '?') {
			i += 3;
			t left = expression(i);
			i++;
			t right = expression(i);
			long long l[3] = { left.x,left.y,left.z };
			long long r[3] = { right.x,right.y,right.z };
			long long ans[3] = { 0,0,0 };
			for (int i = 0; i < 3; i++) {
				for (int j = 0; j < 3; j++) {
					ans[e[i][j]] += l[i] * r[j] % mod;
					ans[a[i][j]] += l[i] * r[j] % mod;
					ans[e[i][j]] %= mod;
					ans[a[i][j]] %= mod;
				}
			}
			i++;
			return t{ ans[0],ans[1],ans[2] };
		}
		else if (s[i] == 'e') {
			i += 3;
			t left = expression(i);
			i++;
			t right = expression(i);
			long long l[3] = { left.x,left.y,left.z };
			long long r[3] = { right.x,right.y,right.z };
			long long ans[3] = { 0,0,0 };
			for (int i = 0; i < 3; i++) {
				for (int j = 0; j < 3; j++) {
					ans[e[i][j]] += l[i] * r[j] % mod;
					ans[e[i][j]] %= mod;
				}
			}
			i++;
			return t{ ans[0],ans[1],ans[2] };
		}
		else if (s[i] == 'a') {
			i += 3;
			t left = expression(i);
			i++;
			t right = expression(i);
			long long l[3] = { left.x,left.y,left.z };
			long long r[3] = { right.x,right.y,right.z };
			long long ans[3] = { 0,0,0 };
			for (int i = 0; i < 3; i++) {
				for (int j = 0; j < 3; j++) {
					ans[a[i][j]] += l[i] * r[j] % mod;
					ans[a[i][j]] %= mod;
				}
			}
			i++;
			return t{ ans[0],ans[1],ans[2] };
		}
	}
	else {
		if (s[i] == '0') {
			i++;
			return { 1,0,0 };
		}
		else if (s[i] == '1') {
			i++;
			return { 0,1,0 };
		}
		else if (s[i] == '2') {
			i++;
			return { 0,0,1 };
		}
		else {
			i++;
			return { 1,1,1 };
		}
	}
}
int main() {
	cin >> s >> k;
	t ans = expression(i);
	if (k == 0)cout << ans.x <<endl;
	if (k == 1)cout << ans.y << endl;
	if (k == 2)cout << ans.z << endl;
	return 0;
}
0