#P1135. 树

题目描述

现在给你一棵共有nn个节点的树。树上每条边有黑色和白色两种颜色。

显然,树上任意不同的两点u,vu, v之间有一条简单路径。记为d(u,v)d(u, v)。称这条路径是好的,当且仅当这条路径至少满足下列条件之一:

  • 路径上颜色全部相同。
  • 路径上黑色边的总量不大于kk

你的任务是求解共有多少条好路径。其中,d(u,v)d(u, v)d(v,u)d(v, u)视为同一条路径。

输入格式

第一行两个整数nnkk,含义同题面描述。

接下来n1n-1行,每行三个整数u,v,cu, v, c,表示uuvv之间有一条连边,当c=1c=1时为黑边;c=0c=0时为白边。

输出格式

一行一个整数,表示答案。

5 1
1 2 1 
2 3 0
3 4 1
4 5 0
8
6 1
1 2 1 
2 3 0
3 4 1
4 5 1
5 6 0
9
5 1
1 2 1
2 3 1
3 4 0
4 5 1
7

数据规模与约定

每组数据点10分,共10组数据。

数据点编号 数据范围
#1~#3 1n102,1k51\le n\le 10^2, 1 \le k \le 5
#4 1n105,k=01\le n \le 10^5, k = 0
#5 1n105,k=11\le n \le 10^5, k = 1
#6~#7 1n103,1k1031\le n\le 10^3, 1 \le k \le 10^3
#8~#10 1n105,1k1031\le n\le 10^5, 1 \le k \le 10^3

大样例

大样例下载