思路:
考虑左边选恰好用i个字符组成长度为n的串,既然左边都不一样了,那么那么右边就的可以随便选了,直接(m-i)^n种。所以问题就转化成了求i个字符(i个字符都出现)构成长度为n的串的个数,设这个为
f(x)
。 我们可知
f(i)=in−∑i−1j=1Cji∗f(j)
式子意思为:从所有的可能中减去只用其中j种字符构成长度为n的串的种类数,那么剩下的就是所求。
然后就是对于左边和右边的计数了。
左边用 1 种,右边用
(m−1)n
种。
左边用 2 种,右边用
(m−2)n
种。
左边用 i 种,右边用
(m−i)n
种。
。。。。。。
#include<stdio.h>
#include <iostream>
#include<string.h>
#include<math.h>
#include<algorithm>
#define eps 1e-8
typedef long long int lli;
using namespace std;
const int maxn = 2222;
const int mod = 1e9+7;
lli c[maxn][maxn];
lli f[maxn];
lli qp(lli a,lli x){
lli ans = 1;a%=mod;
for(;x;x>>=1){
if(x&1) ans = ans*a%mod;
a = a*a % mod;
}
return ans;
}
void ini(){
c[0][0] = c[1][0] = c[1][1] = 1;
for(int i = 2;i < maxn;i++){
c[i][0] = 1;
for(int j = 1;j <= i;j++){
c[i][j] = (c[i-1][j-1]+c[i-1][j]) % mod;
}
}
}
int main(){
// freopen("1011.in","r",stdin);
// freopen("my.txt","w",stdout);
int t,n,m;lli temp;
scanf("%d",&t);
ini();
while(t--){
scanf("%d%d",&n,&m);
lli ans = 0;
for(int i = 1;i <= min(m,n);i++){
f[i] = qp(i,n);
for(int j = 1;j < i;j++){
f[i] = f[i] - c[i][j]*f[j]%mod;
f[i]=(f[i]%mod+mod)%mod;
}
}
for(int i = 1;i <= min(m-1,n);i++){
ans += c[m][i]*f[i]%mod*qp(m-i,n)%mod;
ans %= mod;
}
printf("%lld\n",(ans+mod)%mod);
}
return 0;
}
第二种方法:
用第二类Stirling数解决左边选i个字符组成长度为n的串的个数。
就是省掉了容斥的过程。。
第二类stirling数s(n,i)描述的是:把n个元素,放到i个非空的集合中的方法数。
对于这道题,我们可以把n位放到 i 个字符集合中,字符集合非空代表着一定有一位是这个字符,所以s(n,i)即用了i个字符组成的长度为n的串的个数。
#include<stdio.h>
#include <iostream>
#include<string.h>
#include<math.h>
#include<algorithm>
#define eps 1e-8
typedef long long int lli;
using namespace std;
const int mod = 1e9+7;
const int maxn = 2e3+20;
lli stl[maxn][maxn];
lli c[maxn][maxn];
lli a[maxn];
void stirling2(){
stl[1][1] = 1;
for(lli i = 2;i < maxn;i++){
for(lli j = 1;j <= i;j++){
stl[i][j] = ( (lli)j*stl[i-1][j] + stl[i-1][j-1] )% mod;
}
}
}
inline lli qp(lli a,lli x){
lli ans = 1;a %= mod;
for(;x;x>>=1){
if(x&1) ans = ans*a % mod;
a = a*a % mod;
}
return ans;
}
void ini(){
a[0] = a[1] = 1;
for(int i = 2;i < maxn;i++){
a[i] = a[i-1] * i % mod;
}
c[0][0] = c[1][0] = c[1][1] = 1;
for(int i = 2;i < maxn;i++){
c[i][0] = 1;
for(int j = 1;j <= i;j++){
c[i][j] = (c[i-1][j-1]+c[i-1][j] ) % mod;
}
}
}
int main(){
// freopen("1011.in","r",stdin);
// freopen("my.txt","w",stdout);
int t,n,m;
scanf("%d",&t);
stirling2();
ini();
while(t--){
scanf("%d%d",&n,&m);
lli ans = 0;
for(int i = 1;i <= min(n,m-1);i++){
ans += c[m][i]*stl[n][i]%mod *a[i]%mod * qp(m-i,n)%mod; ans %= mod;
}
printf("%lld\n",(ans+mod)%mod);
}
return 0;
}