洛谷P2593 [ ZJOI 2006 ] 超级麻将 —— DP

题目：https://www.luogu.org/problemnew/show/P2593

DP的话，考虑到当前这一位只跟前两位有关，所以记录一下这3位的状态就行；

于是一开始记录的第 i 位，i-1 位的数量，i-2 位的数量，是否有过对子，然后可行性DP；

但是写得太丑，又WA又T...

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,a[];
bool f[][][][];
int rd()
{
    int ret=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();}
    while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
    return ret*f;
}
int main()
{
    scanf("%d",&n);
    while(n--)
    {
        memset(a,,sizeof a);
        memset(f,,sizeof f);
        for(int i=;i<=;i++)a[i]=rd();
        f[][][][]=;
        for(int i=;i<=;i++)
            for(int j=;j<=(i->?a[i-]:);j++)
                for(int k=j;k<=(i->?a[i-]:);k++)    if(f[i-][j][k][]||f[i-][j][k][])
                {
                    if(a[i]>=j)f[i][k-j][a[i]-j][]=f[i-][j][k][];
                    if(a[i]>=j)f[i][k-j][a[i]-j][]=f[i-][j][k][];
                    if(a[i]-j>=&&f[i-][j][k][])f[i][k-j][a[i]-j-][]=;
                    for(int l=;a[i]-j>=*l;l++)
                    {
                        if(a[i]-j>=*l)
                        {
                            f[i][k-j][a[i]-j-*l][]=f[i-][j][k][];
                            f[i][k-j][a[i]-j-*l][]=f[i-][j][k][];
                        }
                        if(a[i]-j>=*l)
                        {
                            f[i][k-j][a[i]-j-*l][]=f[i-][j][k][];
                            f[i][k-j][a[i]-j-*l][]=f[i-][j][k][];
                        }
                    }
                }
        if(f[][][][])printf("Yes\n");
        else printf("No\n");
    }
    return ;
}

囧

于是放弃这个写法了...

看到TJ，记录的是第 i 位和第 i-1 位的数量，这样就可以同层转移啦！于是简洁好多；

但总感觉复杂度有点不对啊，这样最高不是 10^8 吗...总之加上 register 就卡过去了。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,a[];
bool f[][][][];
int rd()
{
    int ret=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();}
    while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
    return ret*f;
}
int main()
{
    n=rd();
    while(n--)
    {
        memset(f,,sizeof f);
        for(register int i=;i<=;i++)a[i]=rd();
        f[][][][]=;
        for(register int i=;i<=;i++)
            for(register int j=;j<=(i->?a[i-]:);j++)
                for(register int k=;k<=a[i];k++)
                {
                    if(k>=)f[i][j][k][]|=f[i][j][k-][];//可以同层转移囧
                    if(k>=)f[i][j][k][]|=f[i][j][k-][],f[i][j][k][]|=f[i][j][k-][];
                    if(k>=)f[i][j][k][]|=f[i][j][k-][],f[i][j][k][]|=f[i][j][k-][];
                    if(k<=j&&k<=a[i-])f[i][j][k][]|=f[i-][a[i-]-k][j-k][],
                                       f[i][j][k][]|=f[i-][a[i-]-k][j-k][];
                }
        if(f[][a[]][a[]][])printf("Yes\n");
        else printf("No\n");
    }
    return ;
}