nowcoder14604 着色方案

最新推荐文章于 2024-03-27 22:57:21 发布

*ACoder*

最新推荐文章于 2024-03-27 22:57:21 发布

阅读量249

点赞数

分类专栏： # 一般动态规划

本文链接：https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/104195048

版权

一般动态规划专栏收录该内容

113 篇文章 1 订阅

订阅专栏

链接

点击跳转

题解

要想让同色不相邻，我可以在 $d p$ 的过程中加入一维， $dp_{ij}$ 表示使用了前 $i$ 种颜色，整个序列被切成了 $j$ 段的方案数。切段的目的是让同色的分开，切开的断面将来是要与异色的段。

用 $g_{i,j,k}$ 表示 $i$ 段颜色 $A$ 的和 $j$ 段颜色 $B$ 的合成 $k$ 段，且同色的段不能相邻的方案数，注意如果开一个 $n\times c_max \times n$ 的数组会 $m l e$ ，经过分析之后可以发现 $i>\lfloor \frac{n+1}{2} \rfloor$ 或 $k>\lfloor \frac{n+1}{2} \rfloor$ 的状态对答案没有贡献，所以我可以只 $d p$ 到 $n/2+\epsilon$ (其中 $\epsilon$ 是自己随便规定的一个数，我程序里是 $5$ )

那么 $d p$ 方程如下：
$dp_{ij} = \sum_{k=0}^{n/2+\epsilon} dp_{i-1,k} \times g_{k,c_i}$

代码

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 901
#define maxk 31
#define maxc 31
#define cl(x) memset(x,0,sizeof(x))
#define rep(i,a,b) for(i=a;i<=b;i++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
#define de(x) cerr<<#x<<" = "<<x<<endl
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
#define mod 1'000'000'007
int f[maxn/2+10][maxc][maxn/2+10][2], c[maxk], dp[maxk][maxn/2+10];
int main()
{
    ll i, j, k, n, K, T=read();
    while(T--)
    {
        K=read();
        n=0;
        rep(i,1,K)c[i]=read(),n+=c[i];
        rep(i,0,n/2+5)rep(j,0,30)rep(k,1,n/2+5)f[i][j][k][0]=f[i][j][k][1]=0;
        f[1][0][1][0]=f[0][1][1][1]=1;
        rep(i,0,n/2+5)rep(j,0,30)rep(k,1,n/2+5)
        {
            auto &F=f[i][j][k];
            if(i)F[0]+=(f[i-1][j][k-1][0]+f[i-1][j][k-1][1])%mod+f[i-1][j][k][1];
            if(j)F[1]+=(f[i][j-1][k-1][0]+f[i][j-1][k-1][1])%mod+f[i][j-1][k][0];
            F[0]%=mod, F[1]%=mod;
        }
        rep(i,1,K)rep(j,1,n/2+5)dp[i][j]=0;
        dp[0][0]=1;
        rep(i,1,K)rep(j,1,n/2+5)rep(k,0,n/2+5)
        {
            ll t=(f[k][c[i]][j][0]+f[k][c[i]][j][1])%mod;
            (dp[i][j]+=(ll)dp[i-1][k]*t%mod)%=mod;
        }
        printf("%d\n",dp[K][1]);
    }
    return 0;
}