Submission #588267

#	Time	Username	Problem	Language	Result	Execution time	Memory
588267		Bench0310	Linear Garden (IOI08_linear_garden)	C++17	100 / 100	33 ms	18056 KiB

linear_garden

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n,mod;
    cin >> n >> mod;
    auto add=[&](int a,int b)->int{return a+b-(a+b>=mod?mod:0);};
    auto sub=[&](int a,int b)->int{return a-b+(a-b<0?mod:0);};
    string s;
    cin >> s;
    vector<int> a(n+1,0);
    for(int i=1;i<=n;i++) a[i]=(s[i-1]=='P');
    vector<array<int,3>> dp(n+1,{0,0,0});
    dp[0]={1,1,1};
    for(int i=1;i<=n;i++)
    {
        for(int j=0;j<3;j++)
        {
            for(int k=0;k<3;k++)
            {
                if(abs(j-k)==1) dp[i][j]=add(dp[i][j],dp[i-1][k]);
            }
        }
    }
    int res=1;
    auto mv=[&](int l,int r,int p,int c)
    {
        l=min(l,--p);
        if(r-l==1) res=add(res,sub(add(dp[c][0],dp[c][1]),1));
        if(r-l==2) res=add(res,dp[c][p-l]);
    };
    int l=0,r=0,p=0;
    for(int i=1;i<=n;i++)
    {
        if(a[i]==1) mv(l,r,p,n-i);
        if(a[i]==1) r=max(r,++p);
        else l=min(l,--p);
    }
    cout << res << "\n";
    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