#5136. Random Swaps of Balls

0

Random Swaps of Balls

Random Swaps of Balls

题目描述

有 N−1N - 1 个白球和一个黑球。这 NN 个球排成一行,黑球最初位于最左边的位置。 Takahashi 将恰好执行以下操作 KK 次。

每次两次独立地从 11 到 NN(包含端点)中均匀随机选择一个整数。设所选的整数为 aa 和 bb。如果 a≠ba \neq b,则交换从左数第 aa 个和第 bb 个球。

在 KK 次操作后,设黑球位于从左数第 xx 个位置。求 xx 的期望值,模 998244353998244353。

什么是模 $998244353$ 的期望值? 可以证明所求的期望值总是有理数。此外,在本题的约束下,可以证明如果此值表示为最简分数 $\frac{P}{Q}$,则 $Q \not\equiv 0 \pmod{998244353}$。因此存在唯一的整数 $R$ 使得 $R \times Q \equiv P \pmod{998244353}, 0 \leq R

输入格式

输入从标准输入给出,格式如下: NN KK

输出格式

在一行中打印答案。

2 1
499122178
3 2
554580198
4 4
592707587

提示

1≤N≤9982443521 \leq N \leq 998244352 1≤K≤1051 \leq K \leq 10^5在一次操作后,黑球位于第 1 个位置和第 2 个位置的概率均为 12\displaystyle \frac{1}{2}。因此,期望值为 32\displaystyle \frac{3}{2}。

标签: AtCoder|abc360E

来源

AtCoder|abc360E