#ABC222E. 红蓝树

红蓝树

红蓝树

题目描述

给定一棵有 NN 个顶点的树、一个长度为 MM 的数列 A=(A1,,AM)A=(A_1,\ldots,A_M),以及一个整数 KK

顶点编号为 11NN,第 ii 条边连接顶点 UiU_i 和顶点 ViV_i

我们将把这棵树的 N1N-1 条边分别涂成红色或蓝色。在 2N12^{N-1} 种涂色方案中,求满足以下条件的方案数,对 998244353998244353 取模。

条件:

在顶点 A1A_1 上放置一枚棋子,然后按 i=1,,M1i=1,\ldots,M-1 的顺序,将棋子沿最短路从顶点 AiA_i 移动到顶点 Ai+1A_{i+1}。所有这些移动结束后,若 RRBB 分别表示棋子经过红边和蓝边的次数,则 RB=KR-B=K 成立。

输入格式

输入按以下格式从标准输入给出:

NN MM KK
A1A_1 A2A_2 \ldots AMA_M
U1U_1 V1V_1
\vdots
UN1U_{N-1} VN1V_{N-1}

输出格式

输出答案。

样例

4 5 0
2 3 2 1 4
1 2
2 3
3 4
2

如果第 11 条和第 33 条边涂红色、第 22 条边涂蓝色,则棋子经过的红边和蓝边次数如下:

从顶点 22 移动到 33 时,经过 00 条红边和 11 条蓝边;

从顶点 33 移动到 22 时,经过 00 条红边和 11 条蓝边;

从顶点 22 移动到 11 时,经过 11 条红边和 00 条蓝边;

从顶点 11 移动到 44 时,经过 22 条红边和 11 条蓝边;

合计经过 33 条红边和 33 条蓝边,满足条件。

另一种满足条件的方法是第 11 条和第 33 条边涂蓝色、第 22 条边涂红色。除此之外没有其他方案满足条件,所以答案为 22

3 10 10000
1 2 1 2 1 2 2 1 1 2
1 2
1 3
0

也可能不存在满足条件的涂色方案。

10 2 -1
1 10
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
126
5 8 -1
1 4 1 4 2 1 3 5
1 2
4 1
3 1
1 5
2

数据范围

  • 2N10002 \le N \le 1000
  • 2M1002 \le M \le 100
  • K105|K| \le 10^5
  • 1AiN1 \le A_i \le N
  • 1Ui,ViN1 \le U_i, V_i \le N
  • 给定的图是一棵树。
  • 输入中的所有值均为整数。
难度 提高
通过率
尝试 0
已通过 0
ID
2276
类型
传统题
Time Limit
2000ms
Memory Limit
1024MiB
上传者
标签