QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#779080#9536. Athlete Welcome CeremonyZycKWA 1ms8224kbC++143.8kb2024-11-24 17:21:472024-11-24 17:21:48

Judging History

This is the latest submission verdict.

  • [2024-11-24 17:21:48]
  • Judged
  • Verdict: WA
  • Time: 1ms
  • Memory: 8224kb
  • [2024-11-24 17:21:47]
  • Submitted

answer

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;

const int N = 310;
const int mod = 1e9 + 7;

int n, q;
string s;
int c[N];
// LL dp[N][N][N][3];
LL dp[2][N][N][3];
LL sum[N][N];

int main()
{
    cin >> n >> q;
    cin >> s;
    s = " " + s;

    for(int i = 1; i <= n; i ++) c[i] = c[i - 1] + (s[i] == '?');

    for(int i = 1; i <= n; i ++)
    {
        if(i == 1)
        {
            if(s[i] == '?')
            {
                dp[i][1][0][0] = 1;
                dp[i][0][1][1] = 1;
                dp[i][0][0][2] = 1;
            }
            else
            {
                dp[i][0][0][s[i] - 'a'] = 1;
            }
        }
        else
        {
            for(int j = 0; j <= n; j ++)
                for(int k = 0; k <= n; k ++)
                    for(int t = 0; t < 3; t ++)
                        dp[i & 1][j][k][t] = 0;

            for(int j = 0; j <= c[i]; j ++)
                for(int k = 0; k <= c[i] - j; k ++)
                {
                    int cnt = c[i] - j - k;
                    assert(cnt >= 0 && cnt <= c[i]);

                    if(s[i] == 'a')
                    {
                        dp[i & 1][j][k][0] = (dp[(i + 1) & 1][j][k][1] + dp[(i + 1) & 1][j][k][2]) % mod;
                        dp[i & 1][j][k][1] = 0;
                        dp[i & 1][j][k][2] = 0;
                    }
                    else if(s[i] == 'b')
                    {
                        dp[i & 1][j][k][0] = 0;
                        dp[i & 1][j][k][1] = (dp[(i + 1) & 1][j][k][0] + dp[(i + 1) & 1][j][k][2]) % mod;
                        dp[i & 1][j][k][2] = 0;
                    }
                    else if(s[i] == 'c')
                    {
                        dp[i & 1][j][k][0] = 0;
                        dp[i & 1][j][k][1] = 0;
                        dp[i & 1][j][k][2] = (dp[(i + 1) & 1][j][k][0] + dp[(i + 1) & 1][j][k][1]) % mod;
                    }
                    else
                    {
                        for(int t = 0; t < 3; t ++)
                        {
                            LL &v = dp[i & 1][j][k][t];
                            v = 0;
                            for(int la = 0; la < 3; la ++)
                                if(la != t)
                                {
                                    if(t == 0) v = (v + dp[(i + 1) & 1][j - 1][k][la]) % mod;
                                    else if(t == 1) (v = v + dp[(i + 1) & 1][j][k - 1][la]) % mod;
                                    else v = (v + dp[(i + 1) & 1][j][k][la]) % mod;
                                }
                        }
                    }
                }
        }
    }

    // cout << dp[n & 1][1][0][0] << endl;

    memset(sum, 0, sizeof sum);

    for(int a = 0; a <= c[n]; a ++)
    {
        for(int b = 0; b <= c[n] - a; b ++)
        {
            for(int t = 0; t < 3; t ++) sum[a][b] = (sum[a][b] + dp[n & 1][a][b][t]) % mod;
            sum[a][b] = (sum[a][b] + (b == 0 ? 0 : sum[a][b - 1])) % mod;
        }
        for(int b = c[n] - a + 1; b <= n; b ++) sum[a][b] = sum[a][b - 1];
    }

    while(q --)
    {
        LL res = 0;
        int x, y, z;
        cin >> x >> y >> z;

        for(int t = 0; t <= x; t ++)
        {
            int l = c[n] - t - z, r = min(y, c[n] - t);
            if(l > r) continue;

            LL add = sum[t][r] - sum[t][l - 1];
            add = (add % mod + mod) % mod;
            res = (res + add) % mod;
        }

        cout << res << endl;
    }
}

/*

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


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


10 1
acab?cbaca
1 1 2


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


*/

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 8052kb

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: 1ms
memory: 7652kb

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: 0ms
memory: 6608kb

input:

1 1
?
0 1 1

output:

2

result:

ok single line: '2'

Test #4:

score: -100
Wrong Answer
time: 1ms
memory: 8224kb

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
998473010
998473026
1
1
998473009
1
998479570
1
998473026

result:

wrong answer 2nd lines differ - expected: '1', found: '998473010'