P10717 【MX-X1-T5】「KDOI-05」简单的树上问题
题目背景
原题链接:。
题目描述
小 S 有一棵 $n$ 个点的树。每个点上有一个灯泡。
小 S 决定进行 $k$ 次闪灯操作。执行闪灯操作时,他会用电脑主机给每个灯泡发送一次闪灯操作。
然而,小 S 的灯泡是劣质品,只有一部分的灯泡可以收到小 S 的闪灯操作。具体地,第 $j$ 个点上的灯泡有 $p_{i,j}$ 的概率收到小 S 的第 $i$ 次闪灯操作。
好在,小 S 的不同灯泡之间有信息传递功能。具体地,如果一个灯泡在两个收到信息的灯泡的树上最短路径上,这个灯泡也能执行闪灯操作(当然,收到信息的灯泡会执行闪灯操作)。
定义一个灯泡 $i$ 的美丽度为 $a_{i,S}$,其中 $S$ 为这个灯泡执行闪灯操作的操作集合。
定义整棵树的美丽度为每个灯泡美丽度的乘积。求整棵树美丽度的期望,对 $998244353$ 取模。
输入格式
无
输出格式
无
说明/提示
**【样例解释 \#1】**
| 收到信息灯泡集合 | 灯泡美丽度 | 树美丽度 |
|:--:|:--:|:--:|
| $\emptyset$ | $1,1,1$ | $1$ |
| $\{1\}$ | $2,1,1$ | $2$ |
| $\{2\}$ | $1,3,1$ | $3$ |
| $\{3\}$ | $1,1,4$ | $4$ |
| $\{1,2\}$ | $2,3,1$ | $6$ |
| $\{1,3\}$ | $2,3,4$ | $24$ |
| $\{2,3\}$ | $1,3,4$ | $12$ |
| $\{1,2,3\}$ | $2,3,4$ | $24$ |
故美丽度的期望是 $\frac{1+2+3+4+6+24+12+24}{8}=\frac{19}{2}$,对 $998244353$ 取模后为 $499122186$。
**【数据范围】**
**本题采用捆绑测试。**
| 子任务编号 | 分值 | $n\leq$ | $k\leq$ | 特殊性质 |
|:--:|:--:|:--:|:--:|:--:|
| $1$ | $5$ | $20$ | $1$ | 无 |
| $2$ | $10$ | $100$ | $2$ | 第 $i$ 条边连接 $i$ 与 $i+1$ 号节点 |
| $3$ | $5$ | $100$ | $8$ | $p_{i,j}=0$ 或 $p_{i,j}=1$ |
| $4$ | $5$ | $100$ | $8$ | $a_{i,S}=[S=\{1,2,\dots,k\}]$ |
| $5$ | $20$ | $100$ | $8$ | 第 $i$ 条边连接 $i$ 与 $i+1$ 号节点 |
| $6$ | $15$ | $100$ | $6$ | 无 |
| $7$ | $15$ | $100$ | $7$ | 无 |
| $8$ | $10$ | $50$ | $8$ | 无 |
| $9$ | $15$ | $100$ | $8$ | 无 |
对于 $100\%$ 的数据:$1\leq n\leq100$,$1\leq k\leq8$,$1\leq u,v\leq n$,保证给出数据为一棵树,保证其他输入数据均为 $[0,998244353)$ 中的整数。