AT_jsc2024_final_a.Random Descents

通过率:0%

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

给定一个长度为 NN 的正整数序列 A=(A1,A2,⋯ ,AN)A=(A_1,A_2,\cdots,A_N)。定义 S=∑1≤i≤NAiS = \sum_{1 \leq i \leq N} A_i。

长度为 SS 的整数序列 xx 满足以下全部条件时(且仅当如此),称为“good”:

  • xx 的每个元素均在 11 到 NN 之间;
  • 对于每个整数 ii(1≤i≤N1 \leq i \leq N),xx 恰好包含 AiA_i 个 ii。

对于整数序列 x=(x1,x2,⋯ ,xS)x=(x_1,x_2,\cdots,x_S),记满足 xi>xi+1x_i > x_{i+1} 的下标 ii 的个数为 f(x)f(x)。

从所有“good”的整数序列 xx 中等概率随机选取一个,求 f(x)f(x) 的期望,结果对 998244353998244353 取模。

期望值  mod 998244353\bmod{998244353} 定义如下:可以证明要计算的期望一定是有理数,并且在本题的约束下,若将期望表示为最简分数 PQ\frac{P}{Q},有 Q≢0(mod998244353)Q \not\equiv 0 \pmod{998244353}。因此,存在唯一的整数 RR 满足 R×Q≡P(mod998244353),0≤R<998244353R \times Q \equiv P \pmod{998244353}, 0 \leq R < 998244353。请输出该整数 RR。

输入格式

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

N A1 A2 ⋯ ANN\ A_1\ A_2\ \cdots\ A_N

输出格式

输出答案。

输入输出样例

  • 输入#1

    2
    1 2

    输出#1

    665496236
  • 输入#2

    5
    1 1 1 1 1

    输出#2

    2
  • 输入#3

    2
    2024 824

    输出#3

    841217737
  • 输入#4

    10
    19237224 3435339 227368010 836006 153314679 115537801 189694444 76016030 159566203 53238564

    输出#4

    873203747

说明/提示

样例解释 1

“good” 的 xx 有 33 种:

  • x=(1,2,2)x=(1,2,2),f(x)=0f(x)=0
  • x=(2,1,2)x=(2,1,2),f(x)=1f(x)=1
  • x=(2,2,1)x=(2,2,1),f(x)=1f(x)=1

因此 f(x)f(x) 的期望为 2/32/3。

数据范围

  • 2≤N≤2500002 \leq N \leq 250000
  • 1≤Ai1 \leq A_i
  • ∑1≤i≤NAi<998244353\sum_{1 \leq i \leq N} A_i < 998244353
  • 所有输入均为整数。

由 ChatGPT 5 翻译

输入解题思路,AI测评打分。不知道怎么写?

首页