DISTK
Cho một cây có \(N\) đỉnh, các đỉnh được đánh số từ 1 đến \(N\), và một số \(K\).
Hãy đếm xem có bao nhiêu cặp đỉnh \((u,v)\) mà có khoảng cách đúng bằng \(K\).
Khoảng cách được tính bằng số cạnh. Chú ý: cặp \((u,v)\) và cặp \((v, u)\) là như nhau.
Input:
Dòng đầu tiên có 2 số \(N\) và \(K (1 ≤ 𝑁 ≤ 50000, 1 ≤ 𝐾 ≤ 200).\)
Dòng tiếp theo có \(𝑁\) số \(𝑝_1, 𝑝_2, . . 𝑝_𝑁\) thể hiện có đường đi từ đỉnh \(𝑖\) đến \(𝑝_𝑖, 𝑝_1 = 0\) vì đỉnh \(1\) là đỉnh gốc, còn lại \(1 ≤ 𝑝_𝑖 ≤ 𝑁\).
Output:
- Một số duy nhất số cặp đỉnh có khoảng cách \(K\).
Sample Input
5 2
0 1 2 3 2
Sample Output
4
Comments