Đếm đường K

Xem dưới dạng PDF

Gửi bài giải


Điểm: 100
Giới hạn thời gian: 1.0s
Giới hạn bộ nhớ: 256M
đầu vào: stdin
Đầu ra: stdout

Tác giả:
Kiểu bài tập

Trong khu rừng FPTOJ có \(N\) cây được nối với nhau bởi \(N-1\) con đường, mỗi con đường có độ dài w_i. Các nhà thám hiểm muốn biết có bao nhiêu cặp cây \((u, v)\) mà đường đi giữa chúng có tổng độ dài chính xác bằng \(K\).

Yêu cầu: Đếm số cặp đỉnh \((u, v)\) (\(u < v\)) sao cho \(\text{dist}(u, v) = K\).

Đầu vào
  • Dòng đầu tiên chứa hai số nguyên \(N, K\) (\(1 \le N \le 10^5, 1 \le K \le 10^6\)). - N-\(1\) dòng tiếp theo, mỗi dòng chứa ba số nguyên \(u, v, w\) (\(1 \le w \le 10^6\)).
Đầu ra
  • In ra một số nguyên duy nhất là số cặp thỏa mãn.
Ví dụ

Đầu vào:

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

Đầu ra:

2
Giải thích

Các cặp có tổng khoảng cách bằng 4: \((1, 3)\) (\(1+3=4\)) và \((5, 2)\) (\(2+2=4\)). Vậy có 2 cặp.

Subtask
Subtask Điểm Giới hạn
1 30 \(N \le 100\)
2 30 \(N \le 5000\)
3 40 \(N \le 10^5\)

Nhận xét

Không có ý kiến tại thời điểm này.