QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#702348#9536. Athlete Welcome Ceremonyucup-team3661#WA 40ms642760kbC++202.2kb2024-11-02 15:47:532024-11-02 15:47:55

Judging History

This is the latest submission verdict.

  • [2024-11-02 15:47:55]
  • Judged
  • Verdict: WA
  • Time: 40ms
  • Memory: 642760kb
  • [2024-11-02 15:47:53]
  • Submitted

answer

#include<bits/stdc++.h>
using namespace std;
using ll = long long;

const ll mod = 1e9+7;

void add(ll &a , ll b){
	a += b;
	a %= mod;
}

ll dp[301][301][301][3];

int main(){
	int N; cin >> N;
	int Q; cin >> Q;
	string S; cin >> S;
	for(int i = 0; i < 301; i++){
		for(int j = 0; j < 301; j++){
			for(int k = 0; k < 301; k++){
				for(int l = 0; l < 3; l++){
					dp[i][j][k][l] = 0;
				}
			}
		}
	}
	if(S[0] == '?'){
		dp[0][1][0][0] = 1;
		dp[0][0][1][1] = 1;
		dp[0][0][0][2] = 1;
	}
	else{
		if(S[0] == 'a')dp[0][1][0][0] = 1;
		if(S[0] == 'b')dp[0][0][1][1] = 1;
		if(S[0] == 'c')dp[0][0][0][2] = 1;
	}
	for(int i = 1; i < N; i++){
		for(int j = 0; j <= N; j++){
			for(int k = 0; k <= N; k++){
				for(int pre = 0; pre < 3; pre++){
					if(S[i] == '?'){
						if(j-1 >= 0 && pre != 0)add(dp[i][j][k][0] , dp[i-1][j-1][k][pre]);
						if(k-1 >= 0 && pre != 1)add(dp[i][j][k][1] , dp[i-1][j][k-1][pre]);
						if(pre != 2)add(dp[i][j][k][2] , dp[i-1][j][k][pre]);
					}
					else{
						ll d = S[i] - 'a';
						if(d == pre)continue;
						if(d == 0){
							if(j-1 >= 0)add(dp[i][j][k][0] , dp[i-1][j-1][k][pre]);
						}
						if(d == 1){
							if(k-1 >= 0)add(dp[i][j][k][1] , dp[i-1][j][k-1][pre]);
						}
						if(d == 2){
							add(dp[i][j][k][2] , dp[i-1][j][k][pre]);
						}
					}
				}
			}
		}
	}
	vector<vector<ll>> sum(N+1 , vector<ll>(N+1));
	for(int i = 0; i <= N; i++){
		for(int j = 0; j < N; j++){
			add(sum[i][j+1] , sum[i][j]);
			for(int k = 0; k < 3; k++)add(sum[i][j+1] , dp[N-1][i][j][k]);
		}
	}
	int ba = 0, bb = 0, bc = 0;
	for(int i = 0; i < N; i++){
		if(S[i] == 'a')ba++;
		if(S[i] == 'b')bb++;
		if(S[i] == 'c')bc++;
	}
	while(Q--){
		int A,B,C; cin >> A >> B >> C;
		A += ba;
		B += bb;
		C += bc;
		ll ans = 0;
		for(int a = 0; a <= A; a++){
			if(B+1 <= N-a-C)continue;
			add(ans , sum[a][B+1]);
			add(ans , mod - sum[a][max(0 , N-a-C)]);
		}
		cout << ans << endl;
	}
	return 0;
	for(int k = 0; k < N; k++){
		for(int i = 0; i <= N; i++){
			for(int j = 0; j <= N; j++){
				cout << dp[k][i][j][0]+dp[k][i][j][1]+dp[k][i][j][2] << ' ';
			}
			cout << endl;
		}
		cout << endl;
	}
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 27ms
memory: 642760kb

input:

6 3
a?b??c
2 2 2
1 1 1
1 0 2

output:

3
1
1

result:

ok 3 lines

Test #2:

score: 0
Accepted
time: 36ms
memory: 642692kb

input:

6 3
??????
2 2 2
2 3 3
3 3 3

output:

30
72
96

result:

ok 3 lines

Test #3:

score: -100
Wrong Answer
time: 40ms
memory: 642620kb

input:

1 1
?
0 1 1

output:

0

result:

wrong answer 1st lines differ - expected: '2', found: '0'