AT_pakencamp_2024_day2_b Biscuit and Ants
题目描述
以蚂蚁的栖息地著称的 Paken 公园有 $N$ 个蚁巢和 $N-1$ 条沟渠。蚁巢编号为 $1, 2, \ldots, N$,沟渠编号为 $1, 2, \ldots, N-1$。第 $i$ 条沟渠($1 \leq i \leq N-1$)连接巢 $U_i$ 与巢 $V_i$,且是双向的。在这里,任意两个巢都可以通过某些沟渠相互连通。
生物学家 Paken 发现了 Paken 公园蚂蚁特有的行为习性:
- 住在巢 $s$($1 \leq s \leq N$)的蚂蚁只有在与巢 $s$ 直接通过沟渠相连的不少于 $K$ 个巢中存在饼干时,才会从这些巢把饼干搬到巢 $s$。一旦搬入,巢 $s$ 就会被视为有饼干。需要注意的是,这一行为不会让原有巢中的饼干减少。
作为实验,Paken 选择若干个蚁巢,最初在这些巢中投放饼干,并调查最终有饼干的蚁巢数。对于所有从 $N$ 个巢中选择至少一个巢最初投放饼干的方法(共 $2^N-1$ 种),对于每种情况,请计算最终有饼干的蚁巢数,并将所有情况的总和对 $P$ 取模输出。
输入格式
输入通过标准输入给出,格式如下:
> $N$ $K$ $P$ $U_1$ $V_1$ $U_2$ $V_2$ $\vdots$ $U_{N-1}$ $V_{N-1}$
输出格式
请输出答案。
说明/提示
## 小题目
1. ($1$ 分)$K = 1$
2. ($1$ 分)$K = N-1$
3. ($2$ 分)$N \leq 15$
4. ($6$ 分)$U_i = i$,$V_i = i+1$($1 \leq i \leq N-1$)
5. ($35$ 分)$N \leq 2000, K \leq 5$
6. ($15$ 分)$K \leq 5$
7. ($25$ 分)$N \leq 10^5$
8. ($15$ 分)无附加限制
## 样例解释 1
例如,最初在巢 $1, 3$ 投放饼干,则巢 $2$ 的蚂蚁会从巢 $1, 3$($2 \geq K$)搬来饼干。最终 $1,2,3$ 这 $3$ 个巢中有饼干。
本样例符合小题目 $3,4,5,6,7$ 的限制。
## 样例解释 2
本样例符合小题目 $3,5,6,7$ 的限制。
## 样例解释 3
本样例符合小题目 $1,3,5,6,7$ 的限制。
## 样例解释 4
本样例符合小题目 $2,3,5,6,7$ 的限制。
## 样例解释 5
本样例符合小题目 $3,5,6,7$ 的限制。
## 约束条件
- $1 \leq K \leq N \leq 10^6$
- $10^8 \leq P \leq 10^9$
- $P$ 是质数
- $1 \leq U_i < V_i \leq N$($1 \leq i \leq N-1$)
- 给出的图是一棵树
- 所有输入均为整数
由 ChatGPT 5 翻译