[ARC074C] RGB Sequence

题目

题意翻译

n件物品,m个要求,3种颜色,n,m<=300,限制为(li,ri,xi) 要求[li~ri]内不同颜色个数为xi 问满足m条件的方案数

题解&&思路

自己的思路

显然是dp，但是普通对m,n进行dp是转移不了的，又注意到这里只有三种颜色，所以考虑对每一种颜色dp，因为颜色本质不同，即dp[i][j][k]表示从第x=max(i,j,k)个节点开始往左找到的第一个有颜色的点(也就是节点x)在i位置，第二个颜色不同的点在j，第三个颜色不同的点在k，那么这样就很好转移了，考虑第i+1个位置放什么，那么dp[i][j][k]就会对dp[i+1][j][k]与dp[i][i+1][k]与dp[i][j][i+1]有贡献，存一下约束的右端点即可，在计算状态的时候判断即可

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int MAXN = 301;
const ll mod = 1e9 + 7;
int n , m;
ll dp[MAXN][MAXN][MAXN];
struct node{
    int x , l;
    node(){}
    node( int X,  int L ){
        x = X , l = L;
    }
};
vector<node>G[MAXN];
int main(){
    //freopen( "colorful.in" , "r" , stdin );
    //freopen( "colorful.out" , "w" , stdout );
    scanf( "%d%d" , &n , &m );
    for( int i = 1 ; i <= m ; i ++ ){
        int l , r , x;scanf( "%d%d%d" , &l ,&r , &x );
        G[r].push_back( node( x , l ) );
    }
    dp[0][0][0] = 1;
    for( int i = 1 ; i <= n ; i ++){
        for( int j = 0 ; j < i ; j ++ ){
            for( int k = 0 ; k <= j ; k ++ ){
                dp[i][j][k] = ( dp[i-1][j][k] + dp[i][j][k] ) % mod;
                dp[i][i-1][j] = ( dp[i][i-1][j] + dp[i-1][j][k] ) % mod;
                dp[i][i-1][k] = ( dp[i][i-1][k] + dp[i-1][j][k] ) % mod;
                for( int l2 = 0 ; l2 < G[i].size() ; l2 ++ ){
                    int x = G[i][l2].x , l1 = G[i][l2].l , u = 1;
                    if( j >= l1 ) u ++;
                    if( k >= l1 ) u ++;
                    if( u != x ) dp[i][j][k] = 0;
                }
            }
        }
    }
    ll ans = 0;
    for( int j = 0 ; j < n ; j ++ ){
        for( int k = 0 ; k <= j ; k ++ )
            ans = ( ans + dp[n][j][k] ) % mod;
    }
    printf( "%lld" , ans );
    return 0;
}