Âm và Dương
Đề bài
Mô tả
Cho một cây gồm đỉnh được đánh số từ đến , nối bởi cạnh. Mỗi cạnh có một loại: loại hoặc loại .
Một đường đi cân bằng giữa hai đỉnh và () là đường đi trên cây từ đến thỏa mãn:
- Tổng số cạnh loại bằng tổng số cạnh loại trên toàn bộ đường đi.
- Tồn tại một đỉnh trên đường đi (, ) sao cho đoạn từ đến cũng có số cạnh loại bằng số cạnh loại .
Hãy đếm số lượng cặp đỉnh không phân biệt thứ tự có đường đi cân bằng.
Dữ liệu vào
- Dòng đầu tiên chứa số nguyên .
- dòng tiếp theo, mỗi dòng chứa ba số nguyên , , mô tả một cạnh nối đỉnh với đỉnh có loại ().
Dữ liệu ra
- In ra một số nguyên duy nhất: số cặp đỉnh có đường đi cân bằng.
Ràng buộc
- Đồ thị đầu vào là một cây.
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 7 1 2 0 3 1 1 2 4 0 5 2 0 6 3 1 5 7 1 |
1 | Chỉ có cặp thỏa mãn. Đường đi có cạnh loại và cạnh loại (cân bằng). Đỉnh nghỉ : đoạn có cạnh loại và cạnh loại (cân bằng). |
| 7 1 2 0 2 3 1 3 4 0 4 5 1 1 6 0 6 7 1 |
3 | Ba cặp hợp lệ: , , . Ví dụ cặp : đường đi có cạnh loại và cạnh loại . Đỉnh nghỉ : đoạn có cạnh mỗi loại. |
Bình luận