[美团 CodeM 初赛 Round A]二分图染色

OneInDark

于 2020-09-10 22:13:32 发布

阅读量218

点赞数

分类专栏： C++ 数学动态规划

本文链接：https://blog.csdn.net/qq_42101694/article/details/108462241

版权

C++ 同时被 3 个专栏收录

547 篇文章 3 订阅

订阅专栏

数学

280 篇文章 1 订阅

订阅专栏

动态规划

220 篇文章 2 订阅

订阅专栏

题目

传送门 to LOJ

思路

考场思路

将对应关系写出来，分别是 $1R,1B,2R,2B,\dots,nR,nB$ 的对应值（数字为编号，字母为颜色）。

选出 $x$ 个值是出现了两次的，然后塞到里面去。但是可能两点之间有两条边。

容斥，假设有 $r$ 种数字是出现于同数字的不同颜色。那么方案数为

${n\choose x}\sum_{r=0}^{x}(-1)^{r}{x\choose r}\frac{n!}{(n-r)!}\left[\frac{(n-r)!}{(n-x)!}\right]^2$

$=\sum_{r=0}^{x}\frac{(-1)^r(n!)^2(n-r)!}{[(n-x)!]^3r!(x-r)!}$

$=\frac{(n!)^2}{[(n-x)!]^2}\sum_{r=0}^{x}{n-r\choose n-x}\cdot\frac{(-1)^r}{r!}$

只出现了一次的又怎么办呢？剩下 $n - x$ 种数字和 $2 n - 2 x$ 个位置，可以随便放。方案应当是

$f(n-x)=\sum_{i=0}^{n-x}{n-x\choose i}\frac{(2n-2x)!}{(2n-2x-i)!}$

换元可知，答案应当为

$\sum_{x=0}^{n}\sum_{r=0}^{n-x}\sum_{i=0}^{x}\frac{(n!)^2}{(x!)^2}{n-r\choose x}\frac{(-1)^r}{r!}{x\choose i}\frac{(2x)!}{(2x-i)!}$

$=\sum_{r=0}^{n}\frac{(-1)^r}{r!}\sum_{i=0}^{n-r}{n-r\choose i}\sum_{x=i}^{n-r}{n-r-i\choose x-i}\frac{(n!)^2}{(x!)^2}\cdot\frac{(2x)!}{(2x-i)!}$

然后暴毙。

正确思路

假设有 $x$ 条红边、 $y$ 条蓝边，则答案似乎应为

$\left[{n\choose x}{n\choose y}\right]^2x!y!$

但是需要去掉某两个点之间有两条边的情况。设有 $r$ 个这样的点，则

$ans=\sum_{r=0}^{n}(-1)^r\left[{n\choose r}{n-r\choose x}{n-r\choose y}\right]^2\cdot r!\cdot x!\cdot y!$

~~为了简化，去掉了两个求和符号。~~

令 $f(m)=\sum_{x=0}^{m}{m\choose x}^2x!$

则 $\sum_{r=0}^{n}(-1)^r\cdot r!\cdot{n\choose r}^2f(n-r)^2$

$f (m)$ 的现实意义为：在 $m\times m$ 的矩阵上放置任意多个 $\tt rook$ ，使其互相不可达。

考虑如何递推计算。考虑第一行上有一个 $\tt rook$ ，则去掉其所在的行列，剩下的就是 $f (m - 1)$ 。

问题来了，第一行没有 $\tt rook$ 怎么办？难道不能转化成子问题了吗？

但是剩下的是一个 $m\times(m-1)$ 的矩阵，最多只有 $m - 1$ 个 $\tt rook$ ，也就最多只会占用 $(m-1)\times(m-1)$ 的矩阵。所以还是 $f (m - 1)$ 的情况，但是要插入一个空的列，所以是 $m\cdot f(m-1)$ 。

然而事情并不妙，因为 $f (m - 1)$ 未必填满了。有多少个空列，就会被计算多少次。所以推不动。

不妨引入辅助数组 $g (m)$ 表示填 $m\times (m-1)$ 矩阵的情况。那么转移必须枚举第一行，这样就可以回到 $f$ 或者 $g$ 的状态。如果有 $\tt rook$ ，显然是 $g (m - 1)$ ，否则是 $f (m - 1)$ 。至此，我们可以写出

$\begin{cases} f(x)=g(x)+x\cdot f(x-1)\\ g(x)=(x-1)\cdot g(x-1)+f(x-1) \end{cases}$

完成了 $\mathcal O(n)$ 递推，剩下一切好搞。

代码

#include <cstdio>
#include <iostream>
#include <vector>
using namespace std;
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(int x){
	if(x > 9) writeint(x/10);
	putchar((x%10)^48);
}
inline int qkpow(int_ b,int q,int m){
	int ans = 1;
	for(; q; q>>=1,b=b*b%m)
		if(q&1) ans = ans*b%m;
	return ans;
}

const int MaxN = 10000005;
const int Mod = 1e9+7;
int inv[MaxN], f[MaxN], g[MaxN];

int main(){
	int n = readint();

	inv[1] = 1;
	for(int i=2; i<=n; ++i)
		inv[i] = (0ll+Mod-Mod/i)*inv[Mod%i]%Mod;
	
	f[0] = 1, g[0] = 0;
	for(int i=1; i<=n; ++i){
		g[i] = ((i-1ll)*g[i-1]+f[i-1])%Mod;
		f[i] = (g[i]+1ll*i*f[i-1])%Mod;
	}
	int_ c = 1; // C(n,r)
	int_ jc = 1; // r!
	int ans = 0;
	for(int r=0; r<=n; ++r){
		int_ t = c*f[n-r]%Mod;
		t = t*t%Mod*jc%Mod;
		if(r&1) t = Mod-t;
		ans = (ans+t)%Mod;
		c = c*(n-r)%Mod*inv[r+1]%Mod;
		jc = jc*(r+1)%Mod;
	}
	printf("%d\n",ans);
	return 0;
}