Submission #1019385

#	Time	Username	Problem	Language	Result	Execution time	Memory
1019385		ttamx	Linear Garden (IOI08_linear_garden)	C++17	100 / 100	188 ms	14160 KiB

linear_garden

#include<bits/stdc++.h>
 
using namespace std;
 
const int N=1e6+5;
 
int n,m;
int dp[3][3][5],ndp[3][3][5];
string s;
tuple<int,int,int> qr[N];
bool ok[N];
 
int& norm(int &x){
    return x%=m;
}
 
int main(){
    cin >> n >> m;
    cin >> s;
    int mn=0,mx=0,cur=0,ans=1;
    for(int i=0;i<n;i++){
        if(s[i]=='P'){
            qr[n-i-1]={cur+1,mn,mx};
            ok[n-i-1]=true;
        }
        cur+=s[i]=='L'?1:-1;
        mn=min(mn,cur);
        mx=max(mx,cur);
    }
    dp[0][0][2]=1;
    for(int i=1;i<=n;i++){
        if(ok[i-1]){
            auto [val,mn2,mx2]=qr[i-1];
            for(int j=0;j<3;j++){
                for(int k=0;k<3;k++){
                    if(max(mx2,val+k)-min(mn2,val-j)<=2){
                        for(int x=0;x<5;x++){
                            ans+=dp[j][k][x];
                            ans%=m;
                        }
                    }
                }
            }
        }
        for(int j=0;j<3;j++){
            for(int k=0;k<3;k++){
                for(int x=0;x<5;x++){
                    ndp[j][k][x]=0;
                }
            }
        }
        for(int j=0;j<3;j++){
            for(int k=0;k<3;k++){
                for(int x=-2;x<=2;x++){
                    if(x<2){
                        ndp[j][max(k,x+1)][x+3]+=dp[j][k][x+2];
                        ndp[j][max(k,x+1)][x+3]%=m;
                    }
                    if(x>-2){
                        ndp[max(j,-x+1)][k][x+1]+=dp[j][k][x+2];
                        ndp[max(j,-x+1)][k][x+1]%=m;
                    }
                }
            }
        }
        for(int j=0;j<3;j++){
            for(int k=0;k<3;k++){
                for(int x=0;x<5;x++){
                    dp[j][k][x]=ndp[j][k][x];
                }
            }
        }
    }
    cout << ans;
}

• Subtask 1: 2 / 2
• Subtask 2: 2 / 2
• Subtask 3: 4 / 4
• Subtask 4: 4 / 4
• Subtask 5: 4 / 4
• Subtask 6: 4 / 4
• Subtask 7: 4 / 4
• Subtask 8: 4 / 4
• Subtask 9: 4 / 4
• Subtask 10: 4 / 4
• Subtask 11: 4 / 4
• Subtask 12: 5 / 5
• Subtask 13: 5 / 5
• Subtask 14: 5 / 5
• Subtask 15: 5 / 5
• Subtask 16: 5 / 5
• Subtask 17: 5 / 5
• Subtask 18: 5 / 5
• Subtask 19: 5 / 5
• Subtask 20: 5 / 5
• Subtask 21: 5 / 5
• Subtask 22: 5 / 5
• Subtask 23: 5 / 5