CF2182D Christmas Tree Decoration

题目描述

有 $n$ 个人决定一起装饰圣诞树。他们有 $n+1$ 个装饰品盒子,编号从 $0$ 到 $n$。最初,第 $i$ 个盒子里有 $a_i$ 个装饰品。 我们称一个大小为 $n$ 的排列 $p$(长度为 $n$ 的数组,里面每个数字 $1$ 到 $n$ 各出现一次)是公平的,如果可以通过如下过程将所有装饰品都挂在树上: - 第 $p_1$ 个人可以从盒子 $0$ 或盒子 $p_1$ 中取一个装饰品,并挂在树上; - 第 $p_2$ 个人可以从盒子 $0$ 或盒子 $p_2$ 中取一个装饰品,并挂在树上; - 以此类推; - 第 $p_n$ 个人后,轮到第 $p_1$ 个人,如此循环,直到所有装饰品都挂完。 在这个过程中,永远不应该出现这样一种情况:第 $i$ 个人需要挂装饰品,而盒子 $0$ 和盒子 $i$ 都为空。如果无法避免这样的情况,则该排列不是公平排列。如果每轮每个人都能选择从哪个盒子取装饰品,始终不出现上述情况,则该排列是公平排列。 你的任务是计算有多少个公平排列。由于答案可能很大,请输出对 $998244353$ 取模的结果。

输入格式

第一行包含一个整数 $t$($1 \le t \le 5000$),表示测试用例数量。 每个测试用例的第一行包含一个整数 $n$($1 \le n \le 50$)。 第二行包含 $n+1$ 个整数 $a_0, a_1, \dots, a_n$($0 \le a_i \le 10^6$)。

输出格式

对于每个测试用例,输出一行一个整数,表示公平排列的数量,对 $998244353$ 取模。

说明/提示

在第一个样例中,公平的排列是 $[1, 2, 3]$ 和 $[1, 3, 2]$。 让我们更仔细地看看排列 $[1, 3, 2]$ 的装饰过程: - 第 $p_1=1$ 个人从盒子 $1$ 拿一个装饰品; - 第 $p_2=3$ 个人从盒子 $0$ 拿一个装饰品; - 第 $p_3=2$ 个人从盒子 $2$ 拿一个装饰品; - 第 $p_1=1$ 个人再次从盒子 $1$ 拿一个装饰品。 注意,如果第 $p_1=1$ 个人第一步取的是盒子 $0$,那么第 $p_2=3$ 个人接下来就无法进行(因为此时盒子 $0$ 和 $3$ 都为空)。但由于可以避免这种情况,该排列依然是公平的。 由 ChatGPT 5 翻译