【HDU6038】Function-思维+组合数学

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Maxwei_wzj/article/details/82925697

测试地址：Function
题目大意： 给出一个关于 $0$到 $n-1$的置换 $a$，一个关于 $0$到 $m-1$的置换 $b$，求有多少从 $0$到 $n-1$映射到 $0$到 $m-1$的映射 $f$，满足 $f(i)=b_{f(a_i)}$。
做法： 本题需要用到思维+组合数学。
根据题目的要求， $f(a_i)$可以通过置换 $b$成为 $f(i)$，所以我们对于置换 $a$画一个反向循环图（即，从点 $i$可以走到点 $a_i$，反向就是从点 $a_i$走到点 $i$），再对置换 $b$画一个循环图，在两张图中分别选出两个点 $x,y$，代表 $f(x)=y$，那么可以发现，当 $x$走一步， $y$也走一步， $f(x)=y$应该还是成立的。这也就说明， $y$在 $b$中的循环长度，应该是 $x$在 $a$中循环长度的因数，这样才能保证 $f(x)=y$在走的过程中总是成立。而因为初始的 $y$又有 $len(y)$种选择（ $len(y)$即 $y$所在循环的长度），因此我们找到了一种统计答案的方法：对一个 $a$中的环，在 $b$中找到所有环长为这个环长的因数的环，累加 $len(y)$。直接统计因数的贡献比较麻烦，我们反过来考虑每个 $b$中的环，它对环长为 $len(y)$的倍数的环有 $len(y)$贡献，我们只需要一开始算出 $count_i$，表示环长为 $i$的 $b$中的环的个数，就可以 $O(m\log m)$计算贡献了。那么对于 $a$中的每个环，可以计算出这个环的方案数，运用乘法原理把每个环的方案数乘起来就行了。
我傻逼的地方：我又分不清楚 $n$和 $m$了…这错误一次比一次觉得傻逼…
以下是本人代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1000000007;
int n,m,a[100010],b[100010],tot;
ll cnt[100010],tmp[100010];
bool vis[100010];

void find_loop(int *nxt,int v)
{
	while(!vis[v])
	{
		vis[v]=1;tot++;
		v=nxt[v];
	}
}

int main()
{
	int t=0;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		for(int i=0;i<n;i++)
			scanf("%d",&a[i]);
		for(int i=0;i<m;i++)
			scanf("%d",&b[i]);
		
		for(int i=0;i<=m;i++)
			tmp[i]=vis[i]=0;
		for(int i=0;i<m;i++)
			if (!vis[i])
			{
				tot=0;
				find_loop(b,i);
				tmp[tot]++;
			}
		for(int i=1;i<=n;i++)
			cnt[i]=0;
		for(int i=1;i<=m;i++)
			for(int j=1;i*j<=n;j++)
				cnt[i*j]=(cnt[i*j]+tmp[i]*(ll)i)%mod;
		
		ll ans=1;
		for(int i=0;i<n;i++)
			vis[i]=0;
		for(int i=0;i<n;i++)
			if (!vis[i])
			{
				tot=0;
				find_loop(a,i);
				ans=ans*cnt[tot]%mod;
			}
		printf("Case #%d: %lld\n",++t,ans);
	}
	
	return 0; 
}