Đế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

Tý cần giải bài toán sau:

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 \(dist(u, v) = K\).

Định dạng đầu vào

  • Dòng đầu chứa số nguyên \(N\) và \(Q\) (\(N, Q \le 10^5\)).
  • \(N-1\) dòng tiếp theo, mỗi dòng chứa hai số nguyên \(u, v\) mô tả một cạnh.

Định dạng đầu ra

  • In ra kết quả của từng truy vấn trên một dòng.

Ví dụ

Input:

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

Output:

3
Đầ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.