Đếm cặp gần nhau

Xem dưới dạng PDF

Gửi bài giải


Điểm: 100
Giới hạn thời gian: 2.0s
Giới hạn bộ nhớ: 256M
đầu vào: stdin
Đầu ra: stdout

Tác giả:
Kiểu bài tập

Khu rừng FPTOJ có \(N\) cây và \(N-1\) con đường nối chúng. Tèo và bạn của mình muốn biết có bao nhiêu cặp cây \((u, v)\) (\(u < v\)) mà khoảng cách giữa chúng không vượt quá \(K\). Khoảng cách giữa hai cây là số con đường cần đi qua để từ cây này đến cây kia.

Yêu cầu: Đếm số cặp đỉnh \((u, v)\) (\(u < v\)) sao cho \(\text{dist}(u, v) \le K\).

Đầu vào
  • Dòng đầu tiên chứa hai số nguyên \(N, K\) (\(1 \le N \le 20000, 1 \le K \le N\)). - N-\(1\) dòng tiếp theo, mỗi dòng chứa hai số nguyên \(u, v\).
Đầ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 2
1 2
1 3
2 4
2 5

Đầu ra:

5
Giải thích

Các cặp có khoảng cách \(\le 2\): \((1, 2)\), \((1, 3)\), \((2, 3)\), \((2, 4)\), \((2, 5)\). Tổng cộng 5 cặp.

Subtask
Subtask Điểm Giới hạn
1 20 \(N \le 200, K \le N\)
2 30 \(N \le 2000, K \le N\)
3 50 \(N \le 20000, K \le N\)

Nhận xét

Không có ý kiến tại thời điểm này.