HDU 6578. Blank (DP)

题目： http://acm.hdu.edu.cn/showproblem.php?pid=6578

题意：
N个位置，M个限制条件；
每个位置填入{0,1,2,3}中的一个数；
每个限制条件(L,R,X)，要求[L,R]区间内有X种不同的数；
输出方案数。

分析：
设{0,1,2,3}最后出现的位置排完序后为(i,j,k,l)，未出现为0；
f[i][j][k][l]代表(i,j,k,l)这种状态的方案数；
最外层循环由1到n，显然这是最后一维l，因此这一维可以滚动掉。
时间复杂度O(n⁴)，空间复杂度O(n³)

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long llong;
const int tmax=105;
const llong mod=998244353;
int n,m;
struct strict{
    int ll,rr,x;
};
vector<strict> Q[tmax];
strict data[tmax];
llong f[2][tmax][tmax][tmax];
void work()
{
    int p=0;
    f[0][0][0][0]=1;
    for(int i=1;i<=n;i++)
    {
        p^=1;
        for(int j=0;j<i;j++)
            for(int k=j;k<i;k++)
                for(int l=k;l<i;l++)
                {
                    if(k==j&&j!=0) continue;
                    if(l==k&&l!=0) continue;
                    bool ok=true;
                    for(vector<strict>::iterator iter=Q[i].begin();iter!=Q[i].end();iter++)
                    {
                        int sum=1;
                        if(j>=iter->ll) sum++;
                        if(k>=iter->ll) sum++;
                        if(l>=iter->ll) sum++;
                        if(sum!=iter->x) ok=false;
                    }
                    if(!ok)
                    {
                        f[p^1][j][k][l]=0;
                        continue;
                    }
                    f[p][j][k][l]=(f[p][j][k][l]+f[p^1][j][k][l])%mod;
                    f[p][k][l][i]=(f[p][k][l][i]+f[p^1][j][k][l])%mod;
                    f[p][j][l][i]=(f[p][j][l][i]+f[p^1][j][k][l])%mod;
                    f[p][j][k][i]=(f[p][j][k][i]+f[p^1][j][k][l])%mod;
                }
        for(int j=0;j<i;j++)
            for(int k=j;k<i;k++)
                for(int l=k;l<i;l++)
                    f[p^1][j][k][l]=0;
    }
    llong ans=0;
    for(int j=0;j<=n;j++)
        for(int k=j;k<=n;k++)
            for(int l=k;l<=n;l++)
            {
                if(k==j&&j!=0) continue;
                if(l==k&&l!=0) continue;
                ans=(ans+f[p][j][k][l])%mod;
            }
    cout<<ans<<endl;
    return;
}
int main()
{
    int T,i;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&data[i].ll,&data[i].rr,&data[i].x);
            Q[data[i].rr].push_back(data[i]);
        }
        work();
        memset(f,0,sizeof(f));
        for(i=1;i<=n;i++)
            Q[i].clear();
    }
    return 0;
}