NOIp十连测 涂色游戏

【问题描述】
小A 和小B 在做游戏。
他们找到了一个n 行m 列呈网格状的画板。小A 拿出了p 支不同颜色的画笔,开始在上面涂色。看到小A 涂好的画板,小B 觉得颜色太单调了,于是把画板擦干净,希望涂上使它看起来不单调的颜色(当然,每个格子里只能涂一种颜色)。小B 想知道一共有多少种不单调的涂色方案。我们定义一个涂色方案是不单调的,当且仅当任意相邻两列都出现了至少q 种颜色。

题解:

都能看出来这是道矩乘题。但是比较变态。

先不考虑矩阵,状态是f[ i ][ j ],指前i列已经填好,第i列共有j种不同颜色的方案数。

这里需要一个另外的g,用来算将j种颜色填入n个格子的方案数

先来看一下g:(我用的是容斥)

    for(int i=;i<=;i++)
{
g[i]=fast(i,n);//(快速幂)
for(int j=;j<i;j++)
{
g[i] = ((g[i] - g[j]*C[i][j]%MOD)%MOD+MOD)%MOD;
}
}

什么意思?

首先什么都不考虑,n个格子都有i种选择,得到i^n。

但是有个问题,就是原来让他有j种颜色,但是最终不够j。因此还要减掉g[ k ]*C[ j ][ k ]。

这样g就求完了。

然后就是状态转移方程了。

设前一列有k种颜色,当前列有j种颜色。

我分了几种情况:

1.k+j<q。这样无法转移……

2.k>=q或j>=q,这样当前列可以随便选,式子比较简单粗暴:

3.其他情况。这里需要枚举j中与k重合的有多少种。

最后转矩阵乘法。时间复杂度O(n^3*log m)。

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MOD 998244353
#define ll long long
ll C[][],g[];
int n,m,p,q;
ll fast(ll x,int y)
{
ll ret = 1ll;
while(y)
{
if(y&)ret=ret*x%MOD;
x=x*x%MOD;
y>>=;
}
return ret;
}
struct mt
{
ll s[][];
}j0,j1;
mt operator * (mt a,mt b)
{
mt ret;
for(int i=;i<=p;i++)
{
for(int j=;j<=p;j++)
{
ret.s[i][j]=;
for(int k=;k<=p;k++)
{
(ret.s[i][j]+=a.s[i][k]*b.s[k][j]%MOD)%=MOD;
}
}
}
return ret;
}
void init()
{
C[][]=;
for(int i=;i<=;i++)
{
C[i][]=;
for(int j=;j<=i;j++)
{
C[i][j]=(C[i-][j-]+C[i-][j])%MOD;
}
}
for(int i=;i<=;i++)
{
g[i]=fast(i,n);
for(int j=;j<i;j++)
{
g[i] = ((g[i] - g[j]*C[i][j]%MOD)%MOD+MOD)%MOD;
}
}
for(int j=;j<=p;j++)
{
for(int k=;k<=p;k++)
{
if(j+k<q)
{
j0.s[j][k]=;
}else
{
if(k<q&&j<q)
{
for(int x=max(q,j)-k;x<=j&&x+k<=p;x++)
{
(j0.s[j][k]+=C[k][j-x]*C[p-k][x]%MOD*g[j]%MOD)%=MOD;
}
}else
{
j0.s[j][k]=C[p][j]*g[j]%MOD;
}
}
}
}
for(int i=;i<=p;i++)j1.s[i][]=g[i]*C[p][i]%MOD;
}
mt fastt(mt x,int y)
{
mt ret;
ret=x;
y--;
while(y)
{
if(y&)ret=ret*x;
x=x*x;
y>>=;
}
return ret;
}
int main()
{
freopen("color.in","r",stdin);
freopen("color.out","w",stdout);
scanf("%d%d%d%d",&n,&m,&p,&q);
init();
mt ans = fastt(j0,m-);
ans = ans*j1;
ll as = ;
for(int i=;i<=p;i++)
as = (as+ans.s[i][])%MOD;
printf("%lld\n",as);
return ;
}
上一篇:算法改进 | java语言中判断素数


下一篇:VB6 GDI+ 入门教程[2] GDI+初始化