hdu-2049不容易系列之(4)——考新郎
不是很好想,一开始想二维dp来做,没推出来,感觉应该也是可以的
这题做法用dp[i]表示到第i对夫妻,i对夫妻都选错的情况数,递推出后,从输入的n对中选出m队,一个组合数就行
#include<bits/stdc++.h>
using namespace std;
const int maxn = 5e6 + 5;
const double pi = 3.1415927;
const int mod = 1000;
typedef long long ll
ll c[21][21];
// dp[i] 表示有i对夫妻,i对夫妻都选错的情况数
ll dp[55];
void init() {
c[0][0] = 1;
for(int i=1;i<=20;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]);
}
}
dp[1] = 0,dp[2] = 1;
for(int i=3;i<=20;i++) {
// 情况1:已经有i-2对选错,从i-1个人里面选个人换老婆,就有n对错
// 情况2: 已经有i-1对错,第i个人的老婆从前面已经选错的老婆拿一个,也会有i对都选错
dp[i] = dp[i-2] * (i-1) + dp[i-1] * (i-1);
}
}
void solve() {
int n, m;
cin>>n>>m;
ll ans = c[n][m] * dp[m];
cout<<ans<<"\n";
}
int main() {
int T = 1;
cin >> T;
init();
while (T--) {
solve();
}
return 0;
}
ps: 我最早的想法
用dp[i][j]维护到第i对夫妻,有j对选错的情况数,可能转移方程不太对