HDU6143 Killer Names

题目链接

题意

两个长度为 $n$ 的字符串，可以使用种类为 $m$ 的字符对字符串进行填充，问两个串不出现相同字符的填充法有几种

分析

​ 考虑 $dp[i][j]$ 表示长度为 $i$ 的字符串使用恰好 $j$ 种字符进行填充的方法种数。于是状态转移方程为 $dp[i][j] = dp[i-1][j-1] \times (m-j+1) + dp[i-1][j] \times j$ 。即选择一种新字符或者从原字符中选一个。最后遍历求和一个字符串使用 i <script type="math/tex" id="MathJax-Element-7">i</script> 种字符时，另一字符串的构建种类即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define LL long long
#define MAXN 2020
const int mod=1e9+7;
LL pm[MAXN][MAXN];
LL dp[MAXN][MAXN];
int main(){
    for(int i=1;i<MAXN;++i){
        pm[i][0]=1;
        for(int j=1;j<MAXN;++j)
            pm[i][j]=(pm[i][j-1]*i)%mod;
    }
    int T,n,m;
    cin>>T;
    while(T--){
        scanf("%d %d",&n,&m);
        dp[0][0]=1;
        for(int i=1;i<=n;++i)
            for(int j=1;j<m;++j)
                dp[i][j]=((m-j+1)*dp[i-1][j-1]+j*dp[i-1][j])%mod;
        LL ans=0;
        for(int i=1;i<m;++i)
            ans=(ans+dp[n][i]*pm[m-i][n])%mod;
        printf("%I64d\n",ans);
    }
}