Cho một cây có đỉnh, các đỉnh được đánh số từ đến , đỉnh số là gốc, và một số . Hãy đếm xem có bao nhiêu cặp đỉnh mà có khoảng cách đúng bằng . Khoảng cách được tính bằng số cạnh. Chú ý: cặp và cặp là như nhau.
Dữ liệu vào:
Dòng đầu tiên có hai số và ;
Dòng tiếp theo có số thể hiện có đường đi từ đỉnh đến với ( vì đỉnh là gốc cây).
Dữ liệu ra:
Một số duy nhất là số cặp đỉnh có khoảng cách bằng .