Submission #553752

#	Time	Username	Problem	Language	Result	Execution time	Memory
553752		AbdelmagedNour	Linear Garden (IOI08_linear_garden)	C++17	100 / 100	22 ms	14124 KiB

linear_garden

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
using namespace std;
const int MAX=1000005;
int dp[MAX][3],mod;
int add(int a,int b){
    return a+b-(a+b>=mod?mod:0);
}
void calc(int N){
    for(int n=0;n<=N;n++){
        for(int d=0;d<3;d++){
            int&ret=dp[n][d];
            if(!n)ret=1;
            else{
                if(d-1>=0)ret=add(ret,dp[n-1][d-1]);
                if(d+1<3)ret=add(ret,dp[n-1][d+1]);
            }
        }
    }
}
int solve1(int n,int d){
    if(d==-2||d==2)return 0;
    if(!n)return 1;
    int&ret=dp[n][d+1];
    if(~ret)return ret;
    ret=add(solve1(n-1,d-1),solve1(n-1,d+1));
    return ret;
}
int solve2(int n,int d){
    if(d==-1||d==3)return 0;
    if(!n)return 1;
    int&ret=dp[n][d];
    if(~ret)return ret;
    ret=add(solve2(n-1,d-1),solve2(n-1,d+1));
    return ret;
}
int solve3(int n,int d){
    if(d==-3||d==1)return 0;
    if(!n)return 1;
    int&ret=dp[n][-d];
    if(~ret)return ret;
    ret=add(solve3(n-1,d-1),solve3(n-1,d+1));
    return ret;
}
int index(string&s){
    int mn=0,mx=0,d=0,res=0,n=s.size();
    for(int i=0;i<n;i++){
        if(s[i]=='L'){
            mn=min(mn,d-1);
            d--;
        }else{
            int x=0;
            if(mx<=1&&min(mn,d-1)>=-1)res=add(res,solve1(n-i-1,d-1)),x+=solve1(n-i-1,d-1)>=1; // -1 to 1
            if(mx<=2&&min(mn,d-1)>=0)res=add(res,solve2(n-i-1,d-1)),x+=solve2(n-i-1,d-1)>=1;    // 0 to 2
            if(mx<=0&&min(mn,d-1)>=-2)res=add(res,solve3(n-i-1,d-1)),x+=solve3(n-i-1,d-1)>=1;
            if(x>1)res=add(res,mod-1);
            mx=max(mx,d+1);
            d++;
        }
    }
    return add(res,1);
}
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int n;
    string s;
    cin>>n>>mod>>s;
    calc(n);
    cout<<index(s);
    return 0;
}

• 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