結果

問題 No.1677 mæx
ユーザー 沙耶花沙耶花
提出日時 2021-09-10 22:02:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 26 ms / 2,000 ms
コード長 1,446 bytes
コンパイル時間 4,088 ms
コンパイル使用メモリ 251,980 KB
最終ジャッジ日時 2025-01-24 10:48:21
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000000000000000
using Data = array<mint,3>;
string S;
int N;

Data MAX(Data a,Data b){
	Data ret;
	rep(i,3)ret[i] = 0;
	rep(i,3){
		rep(j,3){
			ret[max(i,j)] += a[i] * b[j];
		}
	}
	
	return ret;
}

Data MEX(Data a,Data b){
	Data ret;
	rep(i,3)ret[i] = 0;
	rep(i,3){
		rep(j,3){
			if(i!=0&&j!=0){
				ret[0] += a[i]*b[j];
			}
			else if(i!=1&&j!=1){
				ret[1] += a[i]*b[j];
			}
			else{
				ret[2] += a[i]*b[j];
			}
		}
	}
	
	return ret;
}

Data eq(int &pos){
	char c;
	if(S[pos]=='m'){
		c = S[pos+1];
		pos += 3;
		pos++;
		Data x = eq(pos);
		pos++;
		Data y = eq(pos);
		pos++;
		Data ret;
		rep(i,3){
			ret[i] = 0;
		}
		/*
		cout<<c<<endl;
		rep(i,3){
			cout<<x[i].val()<<',';
		}
		cout<<endl;
		rep(i,3){
			cout<<y[i].val()<<',';
		}
		cout<<endl;
		*/
		if(c!='e'){
			Data temp = MAX(x,y);
			rep(i,3){
				ret[i] += temp[i];
			}
		}
		if(c!='a'){
			Data temp = MEX(x,y);
			rep(i,3){
				ret[i] += temp[i];
			}
		}
		return ret;
	}
	else{
		c = S[pos];
		pos++;
		Data ret;
		if(c=='?'){
			ret = {1,1,1};
		}
		else{
			rep(i,3)ret[i] = 0;
			ret[c-'0']++;
		}
		return ret;
	}
	
}

int main(){
	cin>>S;
	int K;
	cin>>K;
	
	N = S.size();
	int pos = 0;
	Data ans = eq(pos);
	
	cout<<ans[K].val()<<endl;
	
	return 0;
}
0