AT_agc063_e [AGC063E] Child to Parent
题目描述
有一棵有 $N$ 个顶点的有根树,顶点编号为 $1$ 到 $N$。顶点 $1$ 是根,顶点 $i$($2\leq i\leq N$)的父节点是 $P_i$。
给定一个非负整数 $r$ 和一个非负整数列 $A = (A_1, \ldots, A_N)$。你可以对这个数列进行任意次(也可以不进行)如下操作:
- 选择一个满足 $i \geq 2$ 且 $A_i \geq 1$ 的 $i$。将 $A_i$ 变为 $A_i - 1$,并将 $A_{P_i}$ 变为 $A_{P_i} + r$。
请你求出,最终可能得到的不同非负整数列 $A$ 的个数,答案对 $998244353$ 取模。
输入格式
输入以如下格式从标准输入给出。
> $N$ $P_2$ $\ldots$ $P_N$ $r$ $A_1$ $\ldots$ $A_N$
输出格式
输出最终可能得到的不同非负整数列 $A$ 的个数,对 $998244353$ 取模。
说明/提示
## 限制条件
- $2 \leq N \leq 300$
- $1 \leq P_i \leq i-1$($2 \leq i \leq N$)
- $0 \leq r \leq 10^9$
- $0 \leq A_i \leq 10^9$
## 样例解释 1
最终可能得到的 $A$ 有以下 $4$ 种:$(1,1,1)$、$(3,0,1)$、$(3,1,0)$、$(5,0,0)$。
## 样例解释 2
最终可能得到的 $A$ 有以下 $5$ 种:$(1,1,1)$、$(1,2,0)$、$(2,0,1)$、$(2,1,0)$、$(3,0,0)$。
## 样例解释 3
最终可能得到的 $A$ 有以下 $6$ 种:$(1,1,1)$、$(1,3,0)$、$(3,0,1)$、$(3,2,0)$、$(5,1,0)$、$(7,0,0)$。
由 ChatGPT 4.1 翻译