CodeForces 869 C.The Intriguing Obsession（组合数学）

Description

三个点集，点数分别为$a,b,c$，现在可以在这些点之间连边，边长度为$1$，要求不能在两个来自同一集合的点连边，且同一集合两点最短距离至少是$3$，问连边方案数

Input

三个整数$a,b,c(1\le a,b,c\le 5000)$

Output

输出连边方案数，结果模$10^9+7$

Sample Input

1 1 1

Sample Output

8

Solution

为保证同一集合两点最短距离至少是$3$只需任一点不能连向另一个集合的两个点即可，这样以来集合两两连边互不影响，分别求出两个集合的连边方案数然后乘起来即可，对于连边的两个集合$a,b$，枚举这两个集合的连边数$i$，方案数即为$C_a^i\cdot C_b^i\cdot i!$，即从两个集合中分别选 i <script type="math/tex" id="MathJax-Element-10">i</script>个点，然后两两匹配

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=5005;
#define mod 998244353
int fact[maxn],inv[maxn];
void init(int n=5e3)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod; 
}
int C(int n,int m)
{
    if(m<0||m>n)return 0;
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int Solve(int a,int b)
{
    int ans=0;
    for(int i=0;i<=a;i++)add(ans,(ll)C(a,i)*C(b,i)%mod*fact[i]%mod);
    return ans;
}
int main()
{
    init();
    int a[4];
    while(~scanf("%d%d%d",&a[0],&a[1],&a[2]))
    {
        sort(a,a+3);
        int ans=1;
        for(int i=0;i<3;i++)
            for(int j=i+1;j<3;j++)
                ans=(ll)ans*Solve(a[i],a[j])%mod;
        printf("%d\n",ans);
    }
    return 0;
}