#ABC352G. 袜子 3

袜子 3

袜子 3

题目描述

高桥君的抽屉里放着各种颜色的袜子。 袜子的颜色用 11NN 的整数表示,颜色 ii 的袜子有 Ai (2)A_i\ (\ge 2) 只。

他打算通过执行以下操作来决定今天穿的袜子:

从抽屉中随机且等概率地一次取出一只袜子,不断重复,直到已经取出的袜子中能凑出两只同颜色的袜子为止。 取出的袜子不再放回抽屉。

求高桥君从抽屉中取出袜子次数的期望值对 998244353998244353 取模的值。

期望值对 998244353998244353 取模的方法

可以证明,所求期望值一定是有理数。 此外,本问题的约束保证了将期望值表示为既约分数 yx\frac{y}{x} 时,xx 不会被 998244353998244353 整除。 此时,存在唯一的整数 zz(0z9982443520 \le z \le 998244352)满足 xzy(mod998244353)xz \equiv y \pmod{998244353}。求出这个 zz

输入格式

输入按以下格式从标准输入给出:

NN
A1A_1 A2A_2 \dots ANA_N

输出格式

输出答案。

样例

2
2 2
665496238

例如,操作可能按如下方式进行:

从抽屉中取出一只颜色 11 的袜子。抽屉中还剩一只颜色 11 的袜子和两只颜色 22 的袜子。

从抽屉中取出一只颜色 22 的袜子。抽屉中还剩颜色 11、颜色 22 的袜子各一只。

从抽屉中取出一只颜色 11 的袜子。此时取出的袜子中,颜色 11 的有两只、颜色 22 的有一只,可以凑出颜色 11 的袜子一对,操作结束。

在这个例子中,高桥君从抽屉中取出了 33 次袜子。

高桥君从抽屉中取出袜子的次数以 23\frac{2}{3} 的概率为 33 次,以 13\frac{1}{3} 的概率为 22 次,因此所求期望值为 $3 \times \frac{2}{3} + 2 \times \frac{1}{3} = \frac{8}{3} \equiv 665496238 \pmod {998244353}$。

1
352
2
6
1796 905 2768 253 2713 1448
887165507

数据范围

  • 1N3×1051 \le N \le 3 \times 10^5
  • 2Ai30002 \le A_i \le 3000
  • 输入中的所有值均为整数
难度 省选/NOI-
通过率
尝试 0
已通过 0
ID
3290
类型
传统题
Time Limit
3000ms
Memory Limit
1024MiB
上传者
标签