P16916 [JLCPC 2026] 计树
题目描述
给出一个整数 $K$,以及两个长度为 $2^K - 1$ 的数组 $a_1, a_2, \ldots, a_{2^K - 1}$ 和 $b_1, b_2, \ldots, b_{2^K - 1}$。
有一个大小为 $2^K$ 的无向完全图 $G$,点编号范围为 $0 \sim 2^K - 1$。
对于图 $G$ 的一棵生成树 $T$,定义
$$
\begin{aligned}
A(T) &= \prod_{(u,v)\in T} a_{u\oplus v},\\
B(T) &= \sum_{(u,v)\in T} b_{u\oplus v},\\
C(T) &= \bigoplus_{(u,v)\in T} (u\oplus v).
\end{aligned}
$$
你需要对于每个 $0 \le x < 2^K$,求出
$$
\left(\sum_{\substack{T\\ C(T)=x}} A(T)B(T)^p\right) \bmod 998244353,
$$
其中 $p$ 为一个给定的常数。
规定 $0^0=1$。
> 无向完全图表示任意两个不同点之间都有一条无向边;生成树表示选出若干条边,使得所有点连通且不存在环;符号 $\oplus$ 表示按位异或。
输入格式
第一行包含两个整数 $K, p$($1 \le K \le 16$,$0 \le p \le 5$)。
接下来一行包含 $2^K - 1$ 个整数,第 $i$ 个整数表示 $a_i$($0 \le a_i < 998244353$)。
接下来一行包含 $2^K - 1$ 个整数,第 $i$ 个整数表示 $b_i$($0 \le b_i < 998244353$)。
输出格式
一行包含 $2^K$ 个整数,第 $i$ 个整数表示 $x = i - 1$ 时的答案。