uoj#P269. 【清华集训2016】如何优雅地求和
【清华集训2016】如何优雅地求和
$\pi\text{+}e$ 在高中上数学课时经常睡觉,然而每次考试都能 $AK$,这让他的同桌叶子很是膜拜。
某天,老师在课上讲二项分布的题目:
(2010天津,18改)
某射手每次射击击中目标的概率p=1/3, 且各次射击的结果相互独立, 互不影响。
记4次中击中的次数为X, 求X的数学期望与期望的方差。
(答案:4/3,8/9)
老师说:“这个,二项分布的期望就是 $np$,方差就是 $np(1-p)$,不信你们自己课后回去算。”
$\pi\text{+}e$ 一听有附加题,马上打起了精神,三下五除二用了他的黑科技“母函数求导”就轻而易举的解决了这个问题。顺便帮叶子也给科普了一下。
2016年暑假的某天,$\pi\text{+}e$ 突然想起了这件事。他想了想,把原来的问题加强了一下,变成下面这个样子:
有一个多项式函数 $f(x)$,最高次幂为 $x^m$,定义变换 $Q$:
$$Q(f,n,x) = \sum_{k = 0}^{n}f(k){n\choose k}x^k(1 - x) ^{n - k}$$
现在给定函数 $f$ 和 $n,x$,求 $Q(f,n,x)\bmod 998244353$。
然而,众所周知,高考考完是会掉很多智商的。$\pi\text{+}e$ 发现自己已经忘记掉怎么用的黑科技了;他打电话给叶子,叶子也说不记得了。
你能帮帮他吗?
出于某种原因,函数 $f$ 由点值形式给出,即给定 $a_0,a_1,\cdots,a_m$ 共 $m+1$ 个数,$f(x)=a_x$。可以证明该函数唯一。
输入格式
第一行三个整数 $n,m,x$,意义如前所述。
第二行共 $m+1$ 个整数,表示 $a_0,a_1,\cdots,a_m$。
输出格式
输出一行一个数表示答案,请对 $998,244,353$ 取模。
4 1 332748118
0 1
332748119
注意到 $332748118 \equiv \frac{1}{3} \pmod{ 998244353 }$, $332748119 \equiv \frac{4}{3} \pmod{ 998244353 }$, $f(x)=x$, 题目中所求的表达式为:
$$\sum_{k=0}^4 k{4\choose k}\left(\frac{1}{3}\right)^k\left(\frac{2}{3}\right)^{4-k}=\frac{4}{3}$$
此即为题目开头二项分布例题计算期望的表达式。
4 3 12
0 1 8 27
46704
经计算可得 $f(x)=x^3$
样例三
见样例数据下载。
限制与约定
对于所有的测试点,保证 $1 \le n \le 10^{9},\ 1 \le m \le 2 \times 10^{4},\ 0\le a_i,x< 998,244,353$。
测试点 | $n \le$ | $m \le$ | 特殊限制 |
---|---|---|---|
1 | $10^{2}$ | $10^{2}$ | $n=m$ |
2 | $10^{3}$ | $10^{3}$ | |
3 | $10^{4}$ | $10^{2}$ | 无约束 |
4 | $10^{5}$ | ||
5 | $10^{6}$ | ||
6 | $10^{9}$ | $= 1$ | |
7 | $= 2$ | $f(x)=x^2$ | |
8 | $f(x)=x^2-x$ | ||
9 | $= 3$ | 无约束 | |
10,11 | $10$ | ||
12,13,14 | $10^{2}$ | ||
15 | $10^{3}$ | ||
16 | $2,000$ | ||
17 | $4,000$ | ||
18 | $8,000$ | ||
19 | $12,000$ | ||
20 | $2 \times 10^{4}$ |
时间限制:$1\texttt{s}$
空间限制:$512\texttt{MB}$