天天看點

Codeforces-869C The Intriguing Obsession(DP/組合數)

傳送門:CF-869C

題意:有3個集合,其中元素分别有a、b、c個,且元素之間各不相同,元素之間可以連一條邊,連邊後相同集合的元素之間的距離不能小于3(當然相同集合的元素之間不能連邊),問連邊的方案數

題解:

首先隻考慮2個集合的情況:

dp[i][j]:集合A,B的元素分别有i個和j個時的方案數

向集合A新增一個元素時(設這一個元素為i),如果i不與任何點連線,則方案數為dp[i][j]=dp[i-1][j]

如果i+1與集合B中一個點連線,由于元素各不相同,于是從B中選一個元素出來的與i相連的方案數為dp[i-1][j-1]*j

再看3個集合的情況,發現ABC三個集合中,AB之間的連線情況并不影響AC之間的連線情況,是以可以得到,

任何2個集合之間的連線情況不會影響其他兩兩集合連線情況,是以答案就是dp[a][b]*dp[b][c]*dp[a][c]

#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
const int MX = 5005;
const LL mod = 998244353;
LL dp[MX][MX];
int main() {
    freopen("in.txt", "r", stdin);
    for (int i = 0; i <= 5000; i++) dp[0][i] = dp[i][0] = 1;
    for (int i = 1; i <= 5000; i++) {
        for (int j = 1; j <= 5000; j++) {
            dp[i][j] = (dp[i - 1][j] + j * dp[i - 1][j - 1]) % mod;
        }
    }
    int a, b, c;
    cin >> a >> b >> c;
    printf("%I64d\n", dp[a][b]*dp[b][c] % mod * dp[a][c] % mod);
    return 0;
}