[arc074e]RGB Sequence

题目大意

有多少三色序列，满足m个条件，每个条件形如区间颜色种类数为多少。

做法

设dp，f[i,j,k]表示1~i填完了，除了第i个位置填的颜色，其余两种颜色最后一次填在j和k，然后每次处理右端点挂在i的条件，把不符合条件的状态赋值为0。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=300+10,mo=1000000007;
int f[maxn][maxn][maxn],h[maxn],go[maxn],dis[maxn],nxt[maxn];
int i,j,k,l,r,t,n,m,ans,tot;
void add(int x,int y,int z){
    go[++tot]=y;
    dis[tot]=z;
    nxt[tot]=h[x];
    h[x]=tot;
}
int main(){
    scanf("%d%d",&n,&m);
    fo(i,1,m){
        scanf("%d%d%d",&l,&r,&t);
        if (r==1&&t!=1){
            printf("0\n");
            return 0;
        }
        add(r,l,t);
    }
    f[1][0][0]=1;
    fo(i,1,n-1){
        fo(j,0,i)
            fo(k,j,i)
                if (f[i][j][k]){
                    if (j||k){
                        (f[i+1][k][i]+=f[i][j][k])%=mo;
                        (f[i+1][j][i]+=f[i][j][k])%=mo;
                    }
                    else (f[i+1][0][i]+=f[i][j][k])%=mo;
                    (f[i+1][j][k]+=f[i][j][k])%=mo;
                }
        t=h[i+1];
        while (t){
            l=go[t];
            fo(j,0,i)
                fo(k,j,i){
                    r=1;
                    if (j>=l) r++;
                    if (k>=l) r++;
                    if (r!=dis[t]) f[i+1][j][k]=0;
                }
            t=nxt[t];
        }
    }
    fo(j,0,n)
        fo(k,j,n){
            r=1;
            if (j) r++;
            if (k) r++;
            if (r==1) (ans+=(ll)f[n][j][k]*3%mo)%=mo;
            else (ans+=(ll)f[n][j][k]*6%mo)%=mo;
        }
    (ans+=mo)%=mo;
    printf("%d\n",ans);
}