Khoảng cách trong cây
Đề bài
Mô tả
Cho một cây (đồ thị liên thông không có chu trình) gồm đỉnh và một số nguyên dương .
Khoảng cách giữa hai đỉnh của cây là số cạnh trên đường đi ngắn nhất nối chúng.
Hãy đếm số cặp đỉnh phân biệt có khoảng cách đúng bằng . Cặp và được coi là cùng một cặp.
Dữ liệu vào
- Dòng đầu chứa hai số nguyên và .
- dòng tiếp theo, mỗi dòng chứa hai số nguyên và mô tả một cạnh nối hai đỉnh và . Tất cả các cạnh đều khác nhau.
Dữ liệu ra
- In ra một số nguyên duy nhất là số cặp đỉnh có khoảng cách đúng bằng .
Ràng buộc
- ,
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 5 2 1 2 2 3 3 4 2 5 |
4 | Các cặp có khoảng cách 2 là , , và . |
| 5 3 1 2 2 3 3 4 4 5 |
2 | Cây là một đường thẳng . Các cặp có khoảng cách 3 là và . |
Bình luận