傳送門: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;
}