QOJ.ac

QOJ

IDSubmission IDProblemHackerOwnerResultSubmit timeJudge time
#1284#820092#9536. Athlete Welcome CeremonyredbluetulipredbluetulipFailed.2024-12-18 19:51:452024-12-18 19:51:45

Details

Extra Test:

Invalid Input

input:

300 0
??????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????????...

output:


result:

FAIL Integer parameter [name=Q] equals to 0, violates the range [1, 100000] (stdin, line 1)

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#820092#9536. Athlete Welcome CeremonyredbluetulipAC ✓635ms235848kbC++173.1kb2024-12-18 19:32:372024-12-18 19:32:46

answer

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
using ll = long long;
const int N=310;
const ll mod=1e9+7;
int n,q;
ll ans[2][N][N][3];
ll f[N][N][N];
string s;
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin>>n>>q;
    cin>>s;
    if(s[0]=='?'){
        ans[0][1][0][0]=ans[0][0][1][1]=ans[0][0][0][2]=1;
    }
    if(s[0]=='a') ans[0][1][0][0]=1;
    if(s[0]=='b') ans[0][0][1][1]=1;
    if(s[0]=='c') ans[0][0][0][2]=1;
    for(int i=1;i<n;i++)
    {
        if(s[i]=='?')
        {
        for(int j=0;j<=i;j++)
        {
            for(int k=0;k+j<=i;k++)
            {
                ans[i&1][j+1][k][0]=(ans[(i-1)&1][j][k][1]+ans[(i-1)&1][j][k][2])%mod;
                ans[i&1][j][k+1][1]=(ans[(i-1)&1][j][k][0]+ans[(i-1)&1][j][k][2])%mod;
                ans[i&1][j][k][2]=(ans[(i-1)&1][j][k][0]+ans[(i-1)&1][j][k][1])%mod;
            }
        }
        }
        else{
            int a=s[i]=='a',b=s[i]=='b';
            for(int j=0;j<=i;j++)
            {
                for(int k=0;k+j<=i;k++)
                {
                    if(s[i]=='a'){
                        ans[i&1][j+1][k][0]=(ans[(i-1)&1][j][k][1]+ans[(i-1)&1][j][k][2])%mod;
                        ans[i&1][j][k+1][1]=0;
                        ans[i&1][j][k][2]=0;
                    }
                    else if(s[i]=='b'){
                        ans[i&1][j+1][k][0]=0;
                        ans[i&1][j][k+1][1]=(ans[(i-1)&1][j][k][0]+ans[(i-1)&1][j][k][2])%mod;
                        ans[i&1][j][k][2]=0;
                    }
                    else{
                        ans[i&1][j+1][k][0]=0;
                        ans[i&1][j][k+1][1]=0;
                        ans[i&1][j][k][2]=(ans[(i-1)&1][j][k][0]+ans[(i-1)&1][j][k][1])%mod;
                    }
                }
            }
        }
    }
    int cnt1=0,cnt2=0,cnt3=0;
    for(int i=0;i<n;i++){
        if(s[i]=='a') cnt1++;
        else if(s[i]=='b') cnt2++;
        else if(s[i]=='c') cnt3++;
    }
    for(int i=0;i<=n;i++){
        for(int j=0;j+i<=n;j++)
        {
            f[i][j][n-i-j]=(ans[(n-1)&1][i][j][0]+ans[(n-1)&1][i][j][1]+ans[(n-1)&1][i][j][2])%mod;
        }
    }
    for(int p=n+1;p<=3*n;p++)
    {
        for(int i=0;i<=n;i++){
            for(int j=min(n,p-i);p-i-j<=n;j--){
                int minus1=(i>0),minus2=(j>0),minus3=(p-i-j>0);
                int cnt=0;
                f[i][j][p-i-j]=(f[i-minus1][j][p-i-j]+f[i][j-minus2][p-i-j]+f[i][j][p-i-j-minus3])%mod;
                if(minus1&&minus2) f[i][j][p-i-j]=(f[i][j][p-i-j]-f[i-1][j-1][p-i-j]+mod)%mod,cnt++;
                if(minus3&&minus2) f[i][j][p-i-j]=(f[i][j][p-i-j]-f[i][j-1][p-i-j-1]+mod)%mod,cnt++;
                if(minus1&&minus3) f[i][j][p-i-j]=(f[i][j][p-i-j]-f[i-1][j][p-i-j-1]+mod)%mod,cnt++;
                if(cnt>1) f[i][j][p-i-j]=(f[i][j][p-i-j]+f[i-minus1][j-minus2][p-i-j-minus3])%mod;
            }
        }
    }
    while(q--)
    {
        int a,b,c;
        cin>>a>>b>>c;
        a=min(n,cnt1+a),b=min(n,cnt2+b),c=min(cnt3+c,n);
        cout<<f[a][b][c]<<'\n';
    }
}