【学习笔记】[AHOI2018初中组] 球球的排列

solution:

算法一：对于 n<=10 ，直接全排列枚举即可。

算法二：对于 n<=300 且 a[i]=1/2 ，直接分类讨论。

算法三：对于 n<=300 且 a[i] 是质数，问题转化为把 m 种颜色的球排列在一排，满足相邻球的颜色不同。

解法一

考虑 一种一种颜色地放。设 $dp[i][j][k]$ 表示前 $i$ 个球的排列中，满足颜色不等于第 $i$ 个球且同色相邻的有 $j$ 对，颜色等于第 $i$ 个球且相邻的有 $k$ 对时，排列 的方案数。（因为这里不好算重，所以设成排列）。答案为 $dp[n][0][0]$ 。只要做到 $O(1)$ 转移就好了。注意开头结尾可能要特殊讨论。

当 $a[i]=a[i+1]$ 时：

Case 1: 将该球放于 与该球颜色相同的球旁边 。假设有 $k^{\prime }$ 个第 $i$ 种颜色的球，$j^{\prime }$ 个非第 $i$ 种颜色的球，那么有 $2k^{\prime }-k$ 个位置可以填，

1. $(2k^{\prime }-k)dp[i][j][k]\text{->}dp[i+1][j][k+1]$

Case 2: 将该球放于两个同色球之间。有 $j$ 个位置可以填，

1. $j\ast dp[i][j][k]\text{->}dp[i+1][j-1][k]$

Case 3: 将该球放于两个异色球之间。减去前两种情况就是这种情况，有 $i+1-(2k^{\prime }-k)-j$ 个位置可以填，

1. $(i+1-(2k^{\prime }-k)-j)\ast dp[i][j][k]\text{->}dp[i+1][j][k]$

当 $a[i]\ne a[i+1]$ 时：

Case 1： 放于两个颜色相同的之间，

1. $(j+k)\ast dp[i][j][k]\text{->}dp[i+1][j+k-1][0]$

Case 2： 放于两个颜色不同的之间，

1. $(i+1-j-k)\ast dp[i][j][k]\text{->}dp[i+1][j+k][0]$

时间复杂度 $O(n^3)$ 。转移系数很恶心。。。还 tm 卡空间。

结合算法一，得到 $80pts$ 。

算法四：简单推一下，会发现关系具有传递性。如果 $a[i]\ast a[j]$ 是平方数，就向 $(i,j)$ 连双向边。对于在同一连通块里的点不能有相邻。举个例子，可以用并查集维护。最后把每个并查集赋一个颜色，下标 $i$ 的颜色就是它所处并查集的编号。最后按颜色排一个序即可。

得分 $100pts$ 。

code:

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PII pair<int,int>
#define ll long long
using namespace std;
const int mx=305;
const int mod=1e9+7;
int n,a[mx],p[mx],fa[mx],id[mx];
ll dp[2][mx][mx];
bool check(ll x) {
	ll y=sqrt(x);
	return y*y==x;
}
void Mod(ll &x,ll y) {
	y%=mod; x=(x+y)%mod;
}
int find(int x) {
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
void unionset(int x,int y) {
	int u=find(x),v=find(y);
	if(u!=v) fa[u]=v;
}
signed main() { 
//   freopen("data.in","r",stdin);
    scanf("%d",&n); for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) {
    	for(int j=i+1;j<=n;j++) {
    		if(check(1ll*a[i]*a[j])) {
    			unionset(i,j);
			}
		}
	}
	int cnt(0);
	for(int i=1;i<=n;i++) {
		if(!id[find(i)]) id[find(i)]=++cnt;
	}
	for(int i=1;i<=n;i++) {
		a[i]=id[find(i)];
	}
	sort(a+1,a+1+n);
	dp[1][0][0]=1;
	int j2=0,k2=0;
	for(int i=1;i<n;i++) {
		if(a[i]==a[i-1]) k2++;
		else j2+=k2,k2=1;
		for(int j=0;j<=i+1;j++) {
			for(int k=0;k<=i+1-j;k++) {
				dp[i+1&1][j][k]=0;
			}
		}
		if(a[i]!=a[i+1]) {
			for(int j=0;j<=i;j++) {
				for(int k=0;k<=i-j;k++) {
				    ll tmp=dp[i&1][j][k];
				    if(!tmp) continue;
				    if(j+k) Mod(dp[i+1&1][j+k-1][0],(j+k)*tmp);
				    Mod(dp[i+1&1][j+k][0],(i+1-j-k)*tmp);
				}
			}
		}
		else {
			for(int j=0;j<=i;j++) {
				for(int k=0;k<=i-j;k++) {
					ll tmp=dp[i&1][j][k];
					if(!tmp) continue;
					Mod(dp[i+1&1][j][k+1],(2*k2-k)*tmp);
					Mod(dp[i+1&1][j][k],(i+1-(2*k2-k)-j)*tmp);
					if(j) Mod(dp[i+1&1][j-1][k],j*tmp);
				}
			}
		}
	}
	printf("%lld\n",dp[n&1][0][0]);
}

解法二 （upd in 2021/10/22）

需要利用组合数的知识。

code:

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PII pair<int,int>
#define ll long long
using namespace std;
const int mx=305;
const int mod=1e9+7;
int n,a[mx],fa[mx],id[mx],s[mx],sum[mx],cnt;
ll c[mx][mx],fac[mx];
ll dp[mx][mx];
bool check(ll x) {
	ll y=sqrt(x);
	return y*y==x;
}
void Mod(ll &x,ll y) {
	x=(x+y)%mod;
}
int find(int x) {
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
void unionset(int x,int y) {
	int u=find(x),v=find(y);
	if(u!=v) fa[u]=v;
}
inline ll G(int x,int y) {
	return c[x-1][y-1]*fac[x]%mod;
}
void init() {
	fac[0]=1;
	for(int i=1;i<=n;i++) fac[i]=fac[i-1]*i%mod;
    for(int i=0;i<=n;i++) c[i][0]=1;
    for(int i=1;i<=n;i++) {
    	for(int j=1;j<=n;j++) {
    		c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
		}
	}
}
void solve() {
	dp[0][0]=1;
	for(int i=1;i<=cnt;i++) {
		sum[i]=sum[i-1]+s[i];
	}
	for(int i=0;i<cnt;i++) {
		for(int j=0;j<=sum[i];j++) {
			if(dp[i][j]) {
				for(int k=1;k<=s[i+1];k++) {
					for(int l=0;l<=min(j,k);l++) {
						Mod(dp[i+1][j+s[i+1]-k-l],dp[i][j]*c[j][l]%mod*c[sum[i]+1-j][k-l]%mod*G(s[i+1],k)%mod);
					}
				}
			}
		}
	}
}
signed main() { 
//   freopen("data.in","r",stdin);
    scanf("%d",&n); for(int i=1;i<=n;i++) fa[i]=i;
    init();
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) {
    	for(int j=i+1;j<=n;j++) {
    		if(check(1ll*a[i]*a[j])) {
    			unionset(i,j);
			}
		}
	}
	for(int i=1;i<=n;i++) {
		id[find(i)]++;
	}
	for(int i=1;i<=n;i++) {
		if(fa[i]==i) {
			s[++cnt]=id[i];
		}
	}
	solve();
	printf("%lld",dp[cnt][0]);
}

o(n^2) 的神仙算法以后再补把。