容斥定理的思想。
当长度为n时,对于姓所用的字符个数从1~m-1讨论。
f[i]表示长度为n的姓用i个字符时,姓的可能数量。
那么f[i] = in-c[i][i-1]*(i-1)n-c[i][i-2]*(i-2)^n…
因为要保证不重复,当用i个字符时,如果只用了i-1个,那么这种情况在只用i-1个字符时就已经讨论过了,所有要减去,其他的也是这样。
具体过程;
先利用杨辉三角,打表组合数。
利用快速幂取模求i^n。
如何每次算f[i],要减去重复部分。
然后,对于名部分,因为保证了姓不重复,名就可以直接认为是(m-i)^n,这样能保证姓名不重复。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=2005;
ll n,m;
ll c[N][N],f[N];
ll f_pow(ll a,ll b)
{
ll res=1;
while(b)
{
if(b&1)
{
res*=a;
res%=mod;
}
a*=a;
a%=mod;
b>>=1;
}
return res%mod;
}
void init()
{
memset(c,0,sizeof(c));
for(int i=0;i<=2000;i++)
{
c[i][0]=1;
c[i][i]=1;
for(int j=1;j<=i;j++)
c[i][j]=(c[i-1][j-1]%mod+c[i-1][j]%mod)%mod;
}
}
int main()
{
int t;
scanf("%d",&t);
init();//预处理组合数
while(t--)
{
scanf("%lld%lld",&n,&m);
ll ans=0;
memset(f,0,sizeof(f));
for(int i=1;i<m;i++)
{
f[i]=f_pow((ll)i,n);
ll tmp=0;
for(int j=i-1;j>0;j--)
tmp=(tmp+c[i][j]*f[j]%mod)%mod;
f[i]-=tmp;
ans=(ans+((c[m][i]*f[i])%mod)*f_pow(m-i,n)%mod)%mod;
}
printf("%lld\n",ans);
}
return 0;
}