HDU 2049 错排问题dp

题意：

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2049
中文题。

---

思路：

错排dp的经典题。
设dp[i][j]表示1~i个人中有j个人找错了自己的新娘，那么考虑第i个人，有三种可能性：

1. 他找对了自己的新娘，那么应该从dp[i-1][j]转移过来
2. 他找错了自己的新娘，而且是和之前找错的人互换新娘，此时两个都错，不过与他交换的那个人之前已经错了，所以应该从dp[i-1][j-1]转移过来，之前一共有j-1个人排错，因此有j-1种可能性
3. 他找错了自己的新娘，而且是和之前找对的人互换，此时两个都错了，但与他交换的那个人之前没错，所以此时增加了2个错位置，应该从dp[i-1][j-2]转移过来，之前一共有i-1-(j-2)个人排对，因此有i-j+1种可能性

可以得到状态转移方程：
dp[i][j] = dp[i-1][j] + dp[i-1][j-1] * (j-1) + dp[i-1][j-2] * (i-j+1)

其实这题可以直接设dp[i]为i个人错位数位i的方案数，递推求解。

---

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

LL dp[25][25];

void init() {
    for (int i = 0; i <= 20; i++) dp[i][0] = 1;
    for (int i = 1; i <= 20; i++) {
        for (int j = 1; j <= 20; j++) {
            dp[i][j] = dp[i - 1][j];
            if (j >= 1) dp[i][j] += dp[i - 1][j - 1] * (j - 1);
            if (j >= 2) dp[i][j] += dp[i - 1][j - 2] * (i - j + 1);
        }
    }
}

int main() {
    //freopen("in.txt", "r", stdin);
    init();
    int T;
    scanf("%d", &T);
    while (T--) {
        int n, m;
        scanf("%d%d", &n, &m);
        printf("%I64d\n", dp[n][m]);
    }
    return 0;
}