HDU 6143 Killer Names （容斥, 2017 Multi-Univ Training Contest 8）

Problem

要求构造若干名字，名字包括 first name last name 两部分，均需包含 n 个字符，已知有 m 种字符供选择，求最多有多少种不同的构造方法，使得 first name 和 last name 不含相同字符。

Idea

容斥。

枚举 first name 用到的字符为 i 个，则 last name 用到的为 m-i 个 ，此时的方案数为 $C_m^i i^n\times (m-i)^n$ 。

当然，此时存在重复枚举的情况。

考虑左边严格取 i 个，右边从剩下的 m-i 个字符中选择（此时不存在重复枚举），故用容斥去除，最终公式为 $(C_m^i i^n + \sum_{j=1}^{i-1} (-1)^{i-j} C_i^j \cdot j^n) \times (m-i)^n$

HINT: 需要注意此题数据范围较大，最好预处理 $a^b$ 以及 $C_a^b$ 。

Code

#include<bits/stdc++.h>
using namespace std;
const int N = 2000 + 10;
const int mod = 1e9 + 7;
int T, n, m;
long long c[N][N];
long long pm[N][N];
void init() {
    c[0][0]=1;
    for(int i=1;i<N;++i){
        c[i][0]=c[i][i]=1;
        for(int j=1;j<i;++j)
            (c[i][j]=c[i-1][j-1]+c[i-1][j]) %= mod;
    }
}

long long solve() {
    long long ans = 0, tmp;
    for(int i=1;i<m;i++)
    {
        tmp = pm[i][n];
        for(int t=0, j=i-1;j;j--, t=1-t)
            if(t == 0)    tmp = (tmp - c[i][j] * pm[j][n] % mod + mod) % mod;
            else    tmp = (tmp + c[i][j] * pm[j][n] % mod) % mod;
        (ans += tmp * c[m][i] % mod * pm[m-i][n]) %= mod;
    }
    return ans;
}

int main()
{
    init();
    for(int i=1;i<N;i++)
    {
        pm[i][0] = 1;
        for(int j=1;j<N;j++)
            pm[i][j] = pm[i][j-1] * i % mod;
    }
    scanf("%d", &T);
    while(T-- && scanf("%d %d", &n, &m)!=EOF)
    {
        printf("%lld\n", solve());
    }
}