51nod 1829 函数 容斥原理

1829 函数

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

想知道f:A->B这个函数（其中|A|=n, |B|=m）的所有映射关系要使B的每个元素都要被A的一个元素覆盖到。

数字可能很大你只要输出方案数模1,000,000,007即可。

Input

一共一行两个数，n和m。（1<=n,m<=1,000,000)

Output

一共一行包含一个方案数。

Input示例

2 2

Output示例

2

m个小球放n个盒子，不允许有空，容斥原理，有（M!）个盒子空的情况 -（（M-1）！）个盒子空的情况+ 容斥一下，就能得到没有盒子空的情况

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6+100;
const int mod = 1e9+7;
ll fac[N+10],inv[N+10];
ll qpow(ll a,ll b)
{
	ll res=1;
	while(b)
	{
		if(b&1) res=res*a%mod;
		a*=a;
		a%=mod;
		b>>=1;
	}
	return res;
}
void init()
{
	fac[0]=1;
	for(int i=1;i<=N;i++)
		fac[i]=fac[i-1]*i%mod;
	inv[N]=qpow(fac[N],mod-2);
	for(int i=N-1;i>=0;i--)
		inv[i]=inv[i+1]*(i+1)%mod;
}

ll C(int a,int b)
{
	if(!b) return 1;
	return fac[a]*inv[b]%mod*inv[a-b]%mod;
}

int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	init();
	ll ans=0;
	int e=1;
	for(int i=0;i<=m;i++,e*=-1)
		ans=(ans+C(m,i)*qpow(m-i,n)*e%mod+mod)%mod;
	printf("%lld\n",ans );
}