想知道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 );
}