QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#720328#9536. Athlete Welcome CeremonyLanzyWA 833ms341216kbC++235.1kb2024-11-07 11:56:292024-11-07 11:56:30

Judging History

This is the latest submission verdict.

  • [2024-11-07 11:56:30]
  • Judged
  • Verdict: WA
  • Time: 833ms
  • Memory: 341216kb
  • [2024-11-07 11:56:29]
  • Submitted

answer

#include<bits/stdc++.h>
#include<bits/extc++.h>
// #define int long long
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)

using namespace __gnu_pbds;
using namespace std;

const int N=3e2+1,mod=998244353;

int dp[2][N][N][N][3];
int n,m;

long long cnt;

void sol(int cases){
    cin>>n>>m;

    string str;
    cin>>str;
    
    int fs=0;
    for(auto v:str){
        if(v=='?') fs++;
    }


    str=" "+str;



    vector<array<int,5>>L,R;

    int sum=0;
    if(str[1]=='a'){
        dp[1][0][0][0][0]=1;
        L.push_back({1,0,0,0,0});
    }else if(str[1]=='b'){
        dp[1][0][0][0][1]=1;
        L.push_back({1,0,0,0,1});
    }else if(str[1]=='c'){
        dp[1][0][0][0][2]=1;
        L.push_back({1,0,0,0,2});
    }else{
        sum++;
        dp[1][1][0][0][0]=1;
        dp[1][0][1][0][1]=1;
        dp[1][0][0][1][2]=1;
        L.push_back({1,1,0,0,0});
        L.push_back({1,0,1,0,1});
        L.push_back({1,0,0,1,2});
    }


    for(int i=1;i<str.size()-1;++i){
        int u=(i%2),v=((i%2)^1);
        for(int s1=0;s1<=sum;++s1){
            for(int s2=0;s2<=sum-s1;++s2){
                int s3=sum-s1-s2;
                if(str[i+1]=='?'){
                    cnt=0;
                    cnt=(cnt+dp[u][s1][s2][s3][1]+dp[u][s1][s2][s3][2])%mod;
                    dp[v][s1+1][s2][s3][0]=(dp[v][s1+1][s2][s3][0]+cnt)%mod;
                    R.push_back({v,s1+1,s2,s3,0});

                    cnt=0;
                    cnt=(cnt+dp[u][s1][s2][s3][0]+dp[u][s1][s2][s3][2])%mod;
                    dp[v][s1][s2+1][s3][1]=(cnt+dp[v][s1][s2+1][s3][1])%mod;
                    R.push_back({v,s1,s2+1,s3,1});

                    cnt=0;
                    cnt=(cnt+dp[u][s1][s2][s3][0]+dp[u][s1][s2][s3][1]);
                    dp[v][s1][s2][s3+1][2]=(cnt+dp[v][s1][s2][s3+1][2])%mod;
                    R.push_back({v,s1,s2,s3+1,2});
                }else if(str[i+1]=='a'){
                    cnt=0;
                    cnt=(dp[u][s1][s2][s3][1]+dp[u][s1][s2][s3][2])%mod;
                    dp[v][s1][s2][s3][0]=(cnt+dp[v][s1][s2][s3][0])%mod;
                    R.push_back({v,s1,s2,s3,0});
                }else if(str[i+1]=='b'){
                    cnt=0;
                    cnt=(cnt+dp[u][s1][s2][s3][0]+dp[u][s1][s2][s3][2])%mod;
                    dp[v][s1][s2][s3][1]=(cnt+dp[v][s1][s2][s3][1])%mod;
                    R.push_back({v,s1,s2,s3,1});
                }else{
                    cnt=0;
                    cnt=(cnt+dp[u][s1][s2][s3][0]+dp[u][s1][s2][s3][1]);
                    dp[v][s1][s2][s3][2]=(cnt+dp[v][s1][s2][s3][2])%mod;
                    R.push_back({v,s1,s2,s3,2});
                }
            }
        }
        if(str[i+1]=='?'){
            sum++;
        }
        for(auto &[s1,s2,s3,s4,s5]:L) dp[s1][s2][s3][s4][s5]=0;
        L.clear();
        swap(L,R);
    }
    
    int k=(n%2);

    for(int s=0;s<=2;++s){
        for(int i=0;i<N;++i){
            for(int j=0;j<N;++j){
                for(int l=0;l<N;++l){
                    if(i+j+l!=sum) {
                        dp[k][i][j][l][s]=0;
                    }else{
                        // cout<<i<<' '<<j<<' '<<l<<' '<<s<<' '<<dp[k][i][j][l][s]<<'\n';
                    }
                }
            }
        }
    }


    for(int s=0;s<=2;++s){
        for(int i=0;i<N;++i){
            for(int j=0;j<N;++j){
                for(int l=0;l<N;++l){
                    cnt=0;
                    if(i-1>=0)
                        cnt+=dp[k][i-1][j][l][s];

                    if(j-1>=0)
                        cnt+=dp[k][i][j-1][l][s];

                    if(l-1>=0)
                        cnt+=dp[k][i][j][l-1][s];

                    if(i-1>=0&&j-1>=0)
                        cnt-=dp[k][i-1][j-1][l][s];

                    if(i-1>=0&&l-1>=0)
                        cnt-=dp[k][i-1][j][l-1][s];

                    if(j-1>=0&&l-1>=0)
                        cnt-=dp[k][i][j-1][l-1][s];
                    
                    if(i-1>=0&&j-1>=0&&l-1>=0)
                        cnt+=dp[k][i-1][j-1][l-1][s];

                    cnt%=mod;
                    dp[k][i][j][l][s]=(cnt+dp[k][i][j][l][s])%mod;
                }
            }
        }
    }


    while(m--){
        int a,b,c;
        cin>>a>>b>>c;

        if(fs==0){
            cout<<0<<'\n';
            continue;
        }

        long long res=0;
        if(str[n]=='?'||str[n]=='a'){
            res+=dp[k][a][b][c][0];
            res%=mod;
            // cout<<dp[k][a][b][c][0]<<'\n';
        }
        if(str[n]=='?'||str[n]=='b'){
            res+=dp[k][a][b][c][1];
            res%=mod;
            // cout<<dp[k][a][b][c][1]<<'\n';
        }
        if(str[n]=='?'||str[n]=='c'){
            res+=dp[k][a][b][c][2];
            res%=mod;
            // cout<<dp[k][a][b][c][2]<<'\n';
        }
        cout<<res<<'\n';
    }
}
       
signed main(){
    IOS;
    int _=1;
    // cin>>_;
    for(int i=1;i<=_;++i){
        sol(i);
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 833ms
memory: 324664kb

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: 803ms
memory: 325096kb

input:

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

output:

30
72
96

result:

ok 3 lines

Test #3:

score: 0
Accepted
time: 809ms
memory: 323248kb

input:

1 1
?
0 1 1

output:

2

result:

ok single line: '2'

Test #4:

score: 0
Accepted
time: 811ms
memory: 323208kb

input:

10 10
acab?cbaca
0 2 0
1 1 2
4 2 3
1 1 1
3 5 1
0 5 2
2 2 0
1 2 5
4 3 0
1 1 3

output:

0
1
1
1
1
0
1
1
1
1

result:

ok 10 lines

Test #5:

score: 0
Accepted
time: 806ms
memory: 325140kb

input:

10 10
?c?c?cbac?
10 5 1
5 8 7
9 2 6
5 7 1
5 2 6
5 6 5
5 10 3
9 1 10
2 5 9
1 2 9

output:

16
16
11
16
11
16
16
5
11
0

result:

ok 10 lines

Test #6:

score: 0
Accepted
time: 826ms
memory: 325912kb

input:

50 100
?abacbacab?cbcbcb?acabcbabcbcacbababc?caba?acacbca
8 3 8
2 4 8
8 7 3
0 9 2
10 8 7
7 6 5
4 10 2
6 9 3
3 6 6
9 10 8
2 5 8
8 1 0
3 5 0
1 0 6
5 0 8
6 5 5
1 7 9
7 7 10
4 7 5
6 6 4
10 1 2
4 1 7
10 0 8
7 6 3
1 9 1
4 7 2
8 4 0
8 6 1
5 10 4
5 8 2
5 8 4
4 5 9
5 2 1
1 10 9
4 10 1
8 4 3
8 9 9
8 0 1
0 8 0...

output:

8
8
8
0
8
8
6
8
8
8
8
0
0
0
1
8
4
8
8
8
2
4
1
8
1
6
0
2
8
6
8
8
1
4
2
8
8
0
0
8
2
0
8
8
8
4
8
8
8
8
2
0
0
4
8
8
1
8
7
6
7
0
8
8
8
0
4
7
8
4
0
8
0
4
8
8
8
7
8
4
7
2
8
8
8
0
2
2
8
8
8
4
4
0
8
0
8
8
1
1

result:

ok 100 lines

Test #7:

score: 0
Accepted
time: 823ms
memory: 325224kb

input:

50 100
b????????bca?????c?b??ca?acac?b?b???ca?ab???a?a???
35 43 36
12 49 47
7 11 34
38 44 22
42 17 10
49 8 38
18 26 44
6 18 14
28 29 6
48 32 47
29 15 48
1 5 33
24 17 18
10 27 32
19 10 34
2 23 9
14 24 39
46 12 34
9 49 26
21 8 46
43 43 3
31 16 2
8 27 7
24 41 35
17 25 31
0 13 47
24 31 23
33 40 30
36 39...

output:

34272000
31599360
497244
34272000
17637520
12290752
34272000
93044
415832
34272000
34272000
0
34272000
16360704
27933952
0
34272000
33886976
7896832
12290752
718
24
0
34272000
34272000
0
34272000
34272000
34272000
32254720
0
5666944
34256640
34272000
34272000
12290752
30493248
34256640
20630016
0
10...

result:

ok 100 lines

Test #8:

score: 0
Accepted
time: 817ms
memory: 323264kb

input:

100 1000
c?cbababcabacbacbacacbacabcbabababacababcbcab?cbabacbacbcbcacbab?bcabcbcababcacbabacbcb?babcbab?baca
13 11 4
4 17 20
14 5 2
16 14 15
8 12 17
19 5 11
5 17 12
20 7 6
19 10 1
6 5 0
13 1 9
7 17 1
20 4 16
11 12 18
19 2 16
18 1 11
19 16 3
7 1 0
6 9 16
6 9 16
6 20 7
0 16 20
1 2 8
16 5 20
18 14 18
...

output:

16
15
14
16
16
16
16
16
8
2
16
8
16
16
16
16
16
2
16
16
16
0
1
16
16
5
1
5
16
16
16
16
16
15
16
13
16
15
2
16
16
1
8
16
16
16
15
0
16
15
16
16
16
16
8
8
16
16
16
16
16
16
8
16
16
1
8
8
16
16
1
16
1
0
16
2
2
16
7
16
16
8
16
16
16
16
1
16
14
16
16
16
16
5
16
16
14
16
11
16
15
11
2
1
8
16
16
7
16
5
16
...

result:

ok 1000 lines

Test #9:

score: -100
Wrong Answer
time: 823ms
memory: 341216kb

input:

100 1000
?????c??????????????????????????b???a????a?????????????????????????c????????????????????????????????
43 38 20
27 40 32
39 27 33
28 50 43
50 3 46
38 46 14
42 48 10
45 25 28
49 10 49
38 17 42
41 49 22
41 18 44
46 47 25
17 44 35
34 43 22
47 42 32
32 44 40
36 41 24
45 38 45
49 44 18
42 34 32
43...

output:

804458366
-11729846
-366568185
-329470027
10104
-191370671
-540591460
-441631694
-165212614
194151457
-228966316
-724203027
-404028188
183174115
-59070904
-394181589
737118944
281846238
-619217374
-337702981
-39036626
575057371
-612575960
989871351
166230586
907706324
-475210023
-411293741
-93030439...

result:

wrong answer 1st lines differ - expected: '490475656', found: '804458366'