CF1503E 2-Coloring
cjx 组合强。
思路
观察一下题目,不难发现只有当黄色形成如下的单峰时才合法。
(染错色了,将就一下)
其中两座峰的峰顶高度相加等于 \(m\),为了方便统计,我们钦定右边的峰一定在左峰下方的行出现,最后答案乘以二就是最终方案。
发现对于每一边是两个最长不下降子序列拼在一起。
设 \(f[i][j]\) 为长度为 \(i\) 的最长不下降子序列,最高点高度为 \(j\) 的方案数。
再设 \(f'[i][j]\) 为长度为 \(i\) 的最长不下降子序列,仅有一个最高点为 \(j\) 的方案数。
答案就是
\[2\times \sum_{i=1}^n\sum_{j=i+1}^{n}\sum_{k=1}^{m-1} (f[i][k]\times f'[n-i+1][k])\times (f'[j][m-k]\times f[n-j+1][m-k]) \]相当于枚举第一个峰在行 \(i\),第二个峰在行 \(j\),第一个峰高度为 \(k\)。
变幻一下:
\[2\times \sum_{i=1}^n\sum_{k=1}^{m-1}(f[i][k]\times f'[n-i+1][k])\times \sum_{j=i+1}^{n} (f'[j][m-k]\times f[n-j+1][m-k]) \]这样给后一个和式做一个后缀和优化,即可把复杂度优化到 \(O(n^2)\) 级别。
当然有例外是这个样子。
两峰交在一起(必须保证在连续的行且峰顶高度相加大于 \(m\)),也是一种方案。
\[2\times \sum_{i=1}^n\sum_{x=1}^{m-1}\sum_{y=1}^{m-1} (f[i][x]\times \sum_{k=1}^{y-1} f[n-i][k])\times (f[n-i][y]\times\sum_{k=1}^{x-1}f[i][k]) \]相当于枚举左边的峰顶在行 \(i\),左峰高度为 \(x\),右峰高度为 \(y\),同时求和了峰过后的一段不下降子序列的方案。
\[2\times \sum_{i=1}^n\sum_{x=1}^{m-1} (f[i][x]\times\sum_{k=1}^{x-1}f[i][k] )\sum_{y=1}^{m-1} (f[n-i][y]\times\sum_{k=1}^{y-1} f[n-i][k]) \]对于 \(\sum_{k=1}^{x-1}f[i][k]\) 和 \(\sum_{k=1}^{y-1} f[n-i][k]\) 前缀和优化。
然后预处理出所有的
\[f[n-i][y]\times\sum_{k=1}^{y-1} f[n-i][k] \]再次前缀和。
即可在 \(O(n^2)\) 的时间内求出上述和式。
对于 \(f\) 和 \(f'\) 的转移是容易的。
\[f[i][j]=\sum_{k=1}^j f[i-1][k] \]\[f'[i][j]=\sum_{k=1}^{j-1} f[i-1][k] \]初始值 \(f[0][0]=1\)。
其实你看到这里,已经可以使用 dp 做出本题了,但是如果你想进一步了解组合,请移步。
不难发现,我们每次的转移都相当于求一次前缀和,而 \(f[0][0]=1\)。
所以有可以构造这样的生成函数:
\[f[0]=1 \]观察 \(\sum_{i=0} x^i\) 的优雅性质,你会发现 \(1\times \sum_{i=0} x^i\) 相当于求一次前缀和。
而 \((\sum_{i=0} x^i)^2\) 相当于求两次前缀和(\(x^i\) 的系数是第 \(i\) 项的求前缀和后的值)。
由 \(f[0]\) 转移到 \(f[1]\) 的过程相当于乘以 \(\sum_{i=0} x^i\)(一次前缀和),所以有:
\[f[1][j]=[x^j](\sum_{k=0} x^k)^1 \]推广一下,每次 \(f[i]\) 向 \(f[i+1]\) 的转移都是乘以 \(\sum_{i=0} x^i\)(每转移一次求一次前缀和),所以有:
\[f[i][j]=[x^j](\sum_{k=0}x^k)^i \]对于 \(\sum_{i=0} x^i\),的封闭形式为 \((1-x)^{-1}\) 有很多种方法求,这里不再赘述。
所以有:
\[f[i][j]=[x^j](1-x)^{-i} \]使用广义二项式展开 \((1+(-x))^{-i}\),即可得到第 \(j\) 项的系数为:
\[\binom{-i}{j}=(-1)^j\binom{i+j-1}{j} \]关于牛顿二项式系数,已经有完备的公式,本处直接套公式即可,故不再展开讨论。
把系数乘上 \((-x)^j\),有
\[f[i][j]=\binom{i+j-1}{j} \]从比较简单的组合意义考虑,有 \(j\) 个球,有 \(i-1\) 个挡板,可以为空,两个挡板之间的球数相当于相邻两项的差。
那么有:
\[f[i][j]=\binom{i+j-1}{i-1}=\binom{i+j-1}{j} \]在观察一下 \(f'\) 求值的式子,你会发现 \(f'[i][j]=f[i][j-1]\)。
所以其实你也求出 \(f'\) 的组合意义了。
相当于有 \(j\) 个球,\(i-1\) 个挡板,最后一个挡板至少挡一个球,那么就可以投入 \(j-1\) 个球供所有挡板挡。
\[f'[i][j]=\binom{i+j-2}{j-1} \]如果考虑生成函数的话,\(f'[i][j]\) 相当于较 \(f[i][j]\) 向后平移了一个位置,那么可以写成:
\[f'[i][j]=[x^{j-1}](1-x)^{-i} \]或
\[f'[i][j]=[x^j](1-x)^{-i}x \]于是你就可以 愉快 的 AC 本题了。
CODE
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define N 5000
#define mod 998244353
const int maxn=5e3+5;
ll n,m,ans;
ll fac[maxn],inv[maxn],f[maxn];
inline ll ksm(ll x,ll y)
{
ll sum=1;
for(;y;y/=2,x=x*x%mod) if(y&1) sum=sum*x%mod;
return sum;
}
inline ll C(int n,int m)
{
if(n<m||m<0) return 0;
return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;
}
int main()
{
freopen("magic.in","r",stdin);
freopen("magic.out","w",stdout);
for(int i=fac[0]=1;i<=N;i++) fac[i]=fac[i-1]*i%mod;
inv[N]=ksm(fac[N],mod-2);
for(int i=N-1;~i;i--) inv[i]=inv[i+1]*(i+1)%mod;
scanf("%lld%lld",&n,&m);
for(int h=1;h<n;h++)
{
for(int j=m;j>=2;j--)
{
f[j]=(1LL*f[j+1]+1LL*C(j+n-h-2,n-h-1)*C(m-j+n-h,n-h)%mod)%mod;
}
for(int i=m-1;i>=1;i--)
{
ans = (1LL*ans+1LL*C(i+h-1,h)*C(m-i+1+h-2,h-1)%mod*f[i+1]%mod)%mod;
}
}
memset(f,0,sizeof(f));
for(int x=1;x<=m;x++)
{
for(int j=n-1;j>=1;j--)
{
f[j]=(1LL*f[j+1]+1LL*C(m-x+n-j-1,n-j-1)*C(m-x+j-1,j)%mod)%mod;
}
for(int i=1;i<n;i++)
{
ans=(1LL*ans+1LL*C(x+i-1,i)*C(x+n-i-1,n-i-1)%mod*f[n-i+1]%mod)%mod;
}
}
printf("%lld\n",ans*2%mod);
}
标签:Coloring,前缀,sum,times,1LL,CF1503E,ll,mod
From: https://www.cnblogs.com/binbinbjl/p/18364095