[AHOI2018初中组]球球的排列

ZBHRuaRua

已于 2022-01-24 15:57:46 修改

阅读量158

点赞数

分类专栏： ACM 文章标签：动态规划算法 acm竞赛

于 2022-01-23 18:30:44 首次发布

本文链接：https://blog.csdn.net/zbhruarua/article/details/122654483

版权

ACM 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

题目

看了半天题解终于大概会了容斥做法。。。

前置知识：
较好的组合数学能力
容斥原理

首先有一个简单的结论，当 $x * y$ 和 $x * z$ 为完全平方数时，显然 $y * z$ 也为完全平方数。
于是可以简单地通过 $O(n^2)$ 暴力将这些球分块。
问题转化为：已知 $n$ 个球属于 $m$ 个颜色，同种颜色小球不能相邻，问排列方案数。

设 $f (s)$ 为至少有 $s$ 对同色小球相邻时的方案数，则由容斥原理知 $ans=\sum_{i=0}^nf(i)*(-1)^i$

考虑如何求 $f (s)$ 前，我们先假设我们将相邻相同颜色的球合并为一个块，设 $b (i)$ 为颜色为 $i$ 的球有多少个块（也就是说现在我们通过固定每一颜色的块数得到了一种情况），则可得当前情况的答案为
$\frac{(\sum_{i=1}^{m}b(i))!}{\prod_{i=1}^mb(i)!}·\prod_{i=1}^{m}a(i)!·C_{a(i)-1}^{b(i)-1}$
设有 $s$ 对同色小球相邻，则 $\sum_{i=1}^{m}b(i)=n-s$
以两个乘号为分界将式子分为三段，第一段为 $b$ 的可重排列数，第二段为每一颜色内小球的全排列，第三段为每一颜色内对小球进行的分块方案（隔板法）

再转换一下可以变成
$(n-s)!\prod_{i=1}^{m}\frac{C_{a(i)-1}^{b(i)-1}·a(i)!}{b(i)!}$
那么实际上 $f (s)$ 就是考虑了所有不同的 $b (i)$ 分配方法的总答案，我们只需要将上式后面这一坨所有的 $b (i)$ 情况所得到的结果相加再乘以 $(n - s)!$ 就得到了 $f (s)$

假设总共分成了 $n - s$ 个块，我们可以把这些块分配到不同的颜色中去，其实也就是一个DP。
设 $H=\prod_{i=1}^{m}\frac{C_{a(i)-1}^{b(i)-1}·a(i)!}{b(i)!}$
则我们要求 $\sum H$

我们设 $g [i] [j]$ 表示考虑了前 $i$ 种颜色，至少有 $j$ 个块时，不同 $b (i)$ 所得到的 $H$ 的和，即 $\sum H$

状态方程是显然的
$g[i][j]=\sum_{k=1}^{min(a(i),j)}g[i-1][j-k]·\frac{C_{a(i)-1}^{k-1}·a(i)!}{k!}$

于是最终答案便是
$\sum_{s=0}^{n-1}g[m][n-s]·(n-s)!·(-1)^s$

完结撒花
PS：可以分析得到容斥做法复杂度为 $O(n^2)$ ，真的好难，还是直接 $O(n^3)$ DP好理解多了。

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<iomanip>
#include<algorithm>
#define ll long long
using namespace std;
const ll INF=1e9+7;
const ll mod=1e9+7;
const ll maxn=305;
inline ll read()
{
	ll x=0,f=1;char c=getchar();
	while(c<'0'||c>'9')
	{
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		x=x*10+(c-'0');
		c=getchar();
	}
	return x*f;
}
ll g[maxn][maxn];
ll n;
ll d[maxn],siz[maxn],vis[maxn],a[maxn],b[maxn],tot;
ll fac[maxn],C[maxn][maxn],inv[maxn],ans;
inline ll quickpow(ll x,ll tim)
{
	ll res=1;
	while(tim)
	{
		if(tim&1) res=res*x%mod;
		x=x*x%mod;
		tim>>=1;
	}
	return res%mod;
}
inline void init()
{
	fac[0]=1;C[0][0]=1;
	for(ll i=1;i<=300;i++) fac[i]=fac[i-1]*i%mod;
	for(ll i=1;i<=300;i++)
	{
		C[i][0]=1;C[i][i]=1;
		for(ll j=1;j<i;j++) C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
	}
	inv[300]=quickpow(fac[300],mod-2);
	for(ll i=299;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
}
int main()
{
	init();
	n=read();
	for(ll i=1;i<=n;i++) d[i]=read(),siz[i]=1;
	for(ll i=1;i<=n;i++)
	{
		for(ll j=i-1;j>=1;j--)
		{
			if(vis[j]) continue;
			if((ll)sqrt(d[i]*d[j])*(ll)sqrt(d[i]*d[j])==d[i]*d[j])
			{
				vis[j]=1;
				siz[i]+=siz[j];
				break;
			}
		}
	}
	for(ll i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			a[++tot]=siz[i];
		}
	}
	g[0][0]=1;
	for(ll i=1;i<=tot;i++)
	{
		for(ll j=1;j<=n;j++)
		{
			ll len=min(a[i],j);
			for(ll k=1;k<=len;k++)
			{
				g[i][j]=(g[i][j]+g[i-1][j-k]*C[a[i]-1][k-1]%mod*fac[a[i]]%mod*inv[k]%mod)%mod;
			}
		}
	}
	for(ll i=0;i<n;i++)
	{
		ll flag=(i%2)?-1:1;
		ans+=g[tot][n-i]*fac[n-i]%mod*flag;
		ans%=mod;
	}
	ans=(ans+mod)%mod;
	cout<<ans<<endl;
	return 0;
}