题意:给你m个字符,两个n个空的字符串,两个字符串不允许出现相同字符,同一字符串可以出现相同字符,问有多少种选法;
当时反对了队友的方法,结果发现竟然是正解,以为会T……
自己当时还是想复杂了,看了题解发现只需要看其中一个就好了,m给其中一个分i个字符,剩余的m-i个字符可以再另一个字符串中任意出现就是(m-i)^n,这样只需求出来n个空填i种字符有多少种情况了;代码:
#include<cstdio> #include<iostream> #include<cstring> #include<cmath> #include<algorithm> using namespace std; typedef long long LL; const int maxn=(int)2e3 +10; const LL MOD=1e9+7; int c[maxn][maxn]; int p[maxn]; int km(int a,int n) { int base=a,ans=1; while(n) { if(n&1) ans=1LL*ans*base%MOD; base=1LL*base*base%MOD; n>>=1; } return ans; } void init() { for(int i=0;i<maxn;i++) { c[i][0]=1;c[i][i]=1; for(int j=1;j<i;j++) { c[i][j]=(c[i-1][j-1]+c[i-1][j])%MOD; } } } int main() { freopen("input.txt","r",stdin); init(); int t,n,m;;scanf("%d",&t); while(t--) { scanf("%d%d",&n,&m); if(m==1){printf("0 ");continue;} p[1]=1; for(int i=2;i<=n;i++) { p[i]=km(i,n); for(int j=1;j<i;j++) p[i]=(p[i]-1LL*c[i][j]*p[j])%MOD; } int ans=0; int s1=min(n,m-1); for(int i=1;i<=s1;i++) { ans=(ans+1LL*p[i]*c[m][i]%MOD*km(m-i,n))%MOD; } printf("%d ",(ans+MOD)%MOD); } return 0; }