Codeforces Round #439 C. The Intriguing Obsession （组合数）

Problem

有三个岛群，分别有 a, b, c 个岛屿。要求在这三个岛群建桥（可以不建），问有多少种不同的建桥方案（有任意一座桥的任意一个端点不同即视为不同方案）。

要求：每座桥的长记为 1，属于同一岛群的两个岛间最短距离至少为 3 ，或者两岛不存在路径。

Idea

由于同岛群的任意两岛最短距离至少为 3 或不能有路径。则可知，非法路径的连接方案为：

• 同岛群两岛直接连接。
• 同岛群两岛均与另一岛群的某岛连接。

故反向条件为：任意两岛群之间取任意 k (岛群数岛群数k∈[0,min(岛群数1,岛群数2)]) 个点，两两建桥均为合法。三个岛群的总方案数即认为是 (a岛群, b岛群) * (a岛群, c岛群) * (b岛群, c岛群) 。

Code

#include<bits/stdc++.h>using namespace std;const int mod = 998244353;int a, b, c;long long C[5010][5010], fac[5010];void init(){    C[0][0] = 1;    fac[0] = 1;    for(int i=1;i<=5000;i++)    {        (fac[i] = fac[i-1] * i) %= mod;        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 x, long long y){    long long mn = min(x, y);    long long ret = 1;    long long tmp;    for(int i=1;i<=mn;i++)    {        tmp = C[x][i] * C[y][i] % mod * fac[i] % mod;        ret = (ret + tmp) % mod;    }    return ret;}int main(){    init();    scanf("%d %d %d", &a, &b, &c);    long long ans = solve(a, b) * solve(b, c) % mod * solve(a, c) % mod;    printf("%lld\n", ans);}