jozj5945. 【NOIP2018模拟11.02】昆特牌

5945. 【NOIP2018模拟11.02】昆特牌

Description
作为一个资深OIer，你被邀请到位于波兰的CDPR总部参观。但没想到你刚一到就遇到了麻烦。昆特牌的数据库发生了故障。原本昆特牌中有 k种卡牌和n 种阵营，为了平衡，每个阵营拥有的卡牌种数都是相等的，并且每个阵营的数据顺序排列。由于故障，卡牌数据被打乱了，每个阵营现在有ai 种卡牌。因为昆特牌即将迎来重大更新，每种牌的所属阵营并不重要，工程师只想尽快让每个阵营拥有相同数量的卡牌。由于数据库的结构原因，你每单位时间只能将一种牌向左边或右边相邻的一个阵营移动。作为OI选手，这自然是难不倒你，但作为一名卡牌游戏爱好者，你想知道最终的卡牌分布有多少种方案。两种方案不同当且仅当存在一种卡牌，它在两种方案中所属阵营不同。对998244353取模

Input
第一行一个整数T，表示数据组数。
接下来每组数据，第一行一个整数n ，第二行n个数，第i个数为ai ,意义见题目描述

Output
T行，每行一个数表示答案。

Sample Input1
3
3
2 1 3
3
1 2 3
3
3 2 1

Sample Input2
4
3
8 1 0
4
5 0 1 2
4
0 4 0 0
4
1 1 6 0

Sample Output

Sample Output1
3
9
9

样例解释
对于第一组数据，初始为{{1,2}{3}{4,5,6}}
移动结束后为
{{1,2}{3,4}{5,6}},{{1,2}{3,6}{4,5}},{{1,2}{3,5}{4,6}}

Sample Output2
1120
30
24
270

Data Constraint

分析：求均摊纸牌的方案数。
考虑贪心的过程，转移形成了一个DAG。在图上DP即可。

代码

#include <cstdio>
#define ll long long
#define mo 998244353
#define N 1000005
using namespace std;

ll pow[N],ny[N],a[N],sum[N];
int T,n;

ll ksm(ll x, ll y)
{
	ll base = x, r = 1;
	while (y)
	{
		if (y & 1) r = (r * base) % mo;
		base = (base * base) % mo;
		y /= 2;
	}
	return r;
}

ll C(int n, int m) {return pow[n] * ny[m] % mo * ny[n - m] % mo;}

int main()
{
	freopen("gwent.in","r",stdin);
    freopen("gwent.out","w",stdout);
	scanf("%d", &T);
	pow[0] = ny[0] = 1;
	for (int i = 1; i <= N; i++)
	{
		pow[i] = pow[i - 1] * i % mo;
		ny[i] = ksm(pow[i], mo - 2);
	}
	while (T--)
	{
		scanf("%d", &n);
		ll ans = 1;
		for (int i = 1; i <= n; i++) 
		{
			scanf("%lld", &a[i]);
			sum[i] = sum[i - 1] + a[i];
		}
		ll x = sum[n] / n;
		for (int i = 1; i <= n; i++)
		{
			ll now = sum[i] - x * i;
			if (now >= 0)
			{
				ans = (ans * C(a[i], now)) % mo;
				a[i + 1] += now;
				a[i] -= now;
			}
			else
			{
				ll s = x * (i + 1) - sum[i];
				ans = (ans * C(s, -now)) % mo;
			}
		}
		printf("%lld\n", ans);
	}
}