牛客国庆集训派对Day4 I 连通块计数(排列组合)

连通块计数

题意：有$n$条链和一个点作为中心组成的树，求非空连通子树数目。

题解：
包含中心的连通子树数目：${\prod }_{i=1}^n(a_i+1)$(这里画个图就知道了)
不包含中心的连通子树数目：${\sum }_{i=1}^{n}C(a_i+1,2)$

代码

#include<bits/stdc++.h>
typedef long long LL;
const int N = 1e7+10, mod = 998244353;
LL fac[N];

LL pow(LL a,LL n)
{
	LL res = a;
	LL ans = 1;
	while(n){
		if(n & 1)
			ans = ans * res % mod;
		res = res * res % mod;
		n >>= 1;
	}
	return ans % mod;
}

LL Inv(LL n)
{
	return pow(n, mod - 2);
}

LL C(LL m, LL n)
{
	if(m < 0|| m < n)
		return 0;
	return fac[m] * Inv(fac[n]) % mod * Inv(fac[m - n]) % mod;
}
int a[N];
int main()
{
#ifndef ONLINE_JUDGE
#endif
	
	fac[0] = 1;
	for(int i = 1; i < N; ++i){
		fac[i] = fac[i - 1] * i % mod;
	}
	int n,m;
	LL ans = 0;
	std::cin>>n;
	for(int i = 0; i < n; ++i){
		std::cin>>a[i];
		ans += C(a[i]+1,2);
		ans %= mod;
	}
	LL b = 1;
	for(int i = 0; i < n; ++i){
		b *= a[i] + 1;
		b %= mod;
	}
	std::cout<<(ans + b) % mod << std::endl;
	return 0;
}