AtCoder Regular Contest 074 E - RGB Sequence 动态规划

题意

有一个长度为n的序列，现在可以给每个位置染上三种颜色中的一种。有m个限制，第i个限制表示在区间[li,ri]中必须恰好有xi种颜色。问满足所有限制的染色方案数。
n,m<=300

分析

考虑从前往后计数。当到达一个位置时，考虑以当前位置为右端点的限制。
不难发现一个性质，颜色数恰好为1的左端点必然是连续的一段，2，3同理。那么我们就可以设dp[i,j,k]表示做到第i位，在[1,k]范围内的左端点恰好出现3种颜色，在[k+1,j]范围内恰好出现2中颜色，在[j+1,i]范围内恰好出现一种颜色。
转移显然。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#define pb(x) push_back(x)
#define mp(x,y) make_pair(x,y)
using namespace std;

typedef pair<int,int> pi;

const int N=305;
const int MOD=1000000007;

int n,m,f[2][N][N];
vector<pi> vec[N];

void updata(int &x,int y)
{
    x+=y;x-=x>=MOD?MOD:0;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        int l,r,x;scanf("%d%d%d",&l,&r,&x);
        vec[r].pb(mp(l,x));
    }
    int now=0;
    f[0][0][0]=3;
    for (int i=1;i<n;i++)
    {
        for (int j=0;j<=i;j++)
            for (int k=0;k<=j;k++)
            {
                if (!f[now][j][k]) continue;
                for (int l=0;l<vec[i].size();l++)
                {
                    int x=vec[i][l].first,y=vec[i][l].second;
                    if (y==1&&x<=j||y==2&&(x<=k||x>j)||y==3&&x>k) {f[now][j][k]=0;break;}
                }
                if (!f[now][j][k]) continue;
                updata(f[now^1][j][k],f[now][j][k]);
                updata(f[now^1][i][k],f[now][j][k]);
                updata(f[now^1][i][j],f[now][j][k]);
                f[now][j][k]=0;
            }
        now^=1;
    }
    int ans=0;
    for (int i=0;i<=n;i++)
        for (int j=0;j<=i;j++)
        {
            for (int k=0;k<vec[n].size();k++)
            {
                int x=vec[n][k].first,y=vec[n][k].second;
                if (y==1&&x<=i||y==2&&(x<=j||x>i)||y==3&&x>j) {f[now][i][j]=0;break;}
            }
            updata(ans,f[now][i][j]);
        }
    printf("%d",ans);
    return 0;
}