Codeforces Round 572 (Div. 1)
E
Problem from Red Panda
Codeforces
3300
原题
← D→
翻译中…

At Moscow Workshops ICPC team gets a balloon for each problem they solved first. Team MSU Red Panda got so many balloons that they didn't know how to spend them. So they came up with a problem with them.

翻译中…

There are several balloons, not more than $10^6$ in total, each one is colored in one of $k$ colors. We can perform the following operation: choose $k-1$ balloons such that they are of $k-1$ different colors, and recolor them all into remaining color. We can perform this operation any finite number of times (for example, we can only perform the operation if there are at least $k-1$ different colors among current balls).

翻译中…

How many different balloon configurations can we get? Only number of balloons of each color matters, configurations differing only by the order of balloons are counted as equal. As this number can be very large, output it modulo $998244353$.

翻译中…

Input

翻译中…

The first line contains a single integer $k$ ($2 \le k \le 10^5$) —the number of colors.

翻译中…

The second line contains $k$ integers $a_1, a_2, \ldots, a_k$ ($0 \le a_i$) —initial configuration of balloons. $a_i$ is number of balloons of color $i$. The total number of balloons doesn't exceed $10^6$. In other words,

翻译中…

$a_1 + a_2 + a_3 + \ldots + a_k \le 10^6$.

翻译中…

Output

翻译中…

Output number of possible configurations modulo $998244353$.

翻译中…

Note

翻译中…

In the first example, there are $3$ configurations we can get: $[0, 1, 2]$, $[2, 0, 1]$, $[1, 2, 0]$.

翻译中…

In the second example, we can apply the operation not more than once, and possible configurations are: $[1, 1, 1, 1]$, $[0, 0, 0, 4]$, $[0, 0, 4, 0]$, $[0, 4, 0, 0]$, $[4, 0, 0, 0]$.

翻译中…

In the third example, we can't apply any operations, so the only achievable configuration is the starting one.

翻译中…

样例

Input #1
3
0 1 2
Output #1
3
InputCopy OutputCopy
Input #2
4
1 1 1 1
Output #2
5
InputCopy OutputCopy
Input #3
5
0 0 1 2 3
Output #3
1
InputCopy OutputCopy
Input #4
3
2 2 8
Output #4
31
InputCopy OutputCopy
题目信息
题号
1188E
时间限制
1 s
内存限制
256 MB
标签
组合数学

题面与样例版权归 Codeforces 及出题人所有;中文翻译由 AI 生成(CI 管线),如有出入请以原文为准。