大致题意:
完美序列的定义:
- ax→ay 为最短路径
- a1→a2→...→ak 中至少经过一条黑边
给你一颗无根树,其中有黑边和红边,求该树中完美序列的数量。
输入:
输入两个正整数 n,k,其中 n 表示无根数的节点数量,k 是一个完美序列的长度(见上)
接下来 n−1 行 ui,vi,di 分别表示一条边与这条边的颜色,di=0 是红色,di=1 是黑色
输出:
完美序列的数量对 109+7 取模
样例
样例输入
4 4
1 2 1
2 3 1
3 4 1
样例输出
252
数据范围
对于 100% 的数据:
- 1≤n,k≤105
- 1≤ui,vi≤n
- di∈1,0