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);
}
}