CF 869C. The Intriguing Obsession 组合

题意:有三个部落(A,B,C) 每个部落分别有a,b,c个城市,现在开始修路,要求相同部落的城市,他们之间的距离为0或者>=3.
a,b,c<=5000. 问有多少种修建道路的方法?


若x为A城市,y,z为B城市
当x-y连边 显然x不能在和B城市的任何点连边 否则y-z距离为2.
所以,若不同部落间有边,则为一一对应.


若A-B间有m条边 则方法数为:C(a,m)*A(b,m) (先从a中选m个盒子 在从b中选m个球,全排列的放入盒子中)
枚举A-B,A-C,B-C 间各有多少条边 O(n^3) TLE.

枚举前面两个部落的边的个数 预处理∑C(b,i)*A(c,i) [i=0,min(b,c)]即可.

...SB了,直接O(N)算出A-B边的总方法数 然后这三段直接相乘即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e3+1,mod=998244353;
int c[N][N],a[N][N];
ll f[N];
void init()
{
	
	f[0]=a[0][0]=c[0][0]=1;
	for(int i=1;i<N;i++)
	{
		f[i]=(f[i-1]*i)%mod;
		c[i][0]=a[i][0]=1;
		for(int j=1;j<=i;j++)
		{
			c[i][j]=(c[i-1][j]+c[i-1][j-1])%mod;
			a[i][j]=(c[i][j]*f[j])%mod;
		}
	}
}
int main()
{
	init();
	int A,B,C;
	scanf("%d%d%d",&A,&B,&C);
	ll res=0,pt=0;
	int t=min(B,C),t1=min(A,B),t2=min(A,C);
	for(int i=0;i<=t;i++)
		pt=(pt+((ll)c[B][i]*(ll)a[C][i])%mod)%mod;	
	for(int i=0;i<=t1;i++)
	{
		ll fi=((ll)c[A][i]*a[B][i])%mod;
		fi=(fi*pt)%mod;
		for(int j=0;j<=t2;j++)
		{
			ll se=((ll)c[A][j]*a[C][j])%mod;
			ll now=(((fi*se)%mod))%mod;
			res=(res+now)%mod;
		}
	}
	printf("%I64d\n",res);
	return 0;
}