Codeforces Round #533 (Div. 2) C. Ayoub and Lost Array（dp）

【题目】

C. Ayoub and Lost Array

【...】

我我我回来啦。

前两题做的都比较顺利，到了这题就有点不稳了，一直都比较畏惧这种问你有多少种情况呀的题目。学长都看不下去了看了看题说是简单dp，点拨了一下，然后我：？？？？？？老子就是不会dp嘛！！！磨了差不多三十分钟的样子样例过不了，以为是我思路做法出了问题，后来才发现初始化的时候dp[1][1]写成dp[1][2]了/手动微笑脸/，但是已经过时间了究极可惜。又要掉分啦。

思路就是定义一个数组保存区间[l,r]内对3取余为0/1/2的数目，然后定义dp[i][j]存前i个元素之和对3取余结果为j的情况数目。状态转移方程见下面的代码。

刚刚去试了下过了。太菜了好了明天开始要继续练题啦。 

【代码】

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int mod=1e9+7;
ll dp[200005][5]={0};
int main()
{
    int n,l,r; cin>>n>>l>>r;
    ll a[4]={0};
    int i=l;
    if(l==r) a[l%3]++;
    else
    {
        for(;i%3!=1;i++) a[i%3]++;
        for(int j=0;j<3;j++) a[j]+=(r-i+1)/3;
        for(i=r/3*3+1;i<=r;i++) a[i%3]++;
    }
    //for(int i=0;i<3;i++) cout<<a[i]<<endl;
    dp[1][0]=a[0],dp[1][1]=a[1],dp[1][2]=a[2];
    for(int i=2;i<=n;i++)   //状态转移方程
    {
        dp[i][0]=(dp[i-1][0]*a[0]+dp[i-1][1]*a[2]+dp[i-1][2]*a[1])%mod;
        dp[i][1]=(dp[i-1][0]*a[1]+dp[i-1][1]*a[0]+dp[i-1][2]*a[2])%mod;
        dp[i][2]=(dp[i-1][0]*a[2]+dp[i-1][1]*a[1]+dp[i-1][2]*a[0])%mod;
    }
    cout<<dp[n][0]<<endl;
    return 0;
}