DHBB 2017 - CNT - 11 - Cặp đôi hoàn hảo
Xem dạng PDFTrong trường hợp đề bài hiển thị không chính xác, bạn có thể tải đề bài tại đây: Đề bài
XYZ là một công ty lớn trong lĩnh vực công nghệ phần mềm. XYZ gồm ~n~ nhân viên, các nhân viên được đánh số từ 1 đến ~n~, nhân viên thứ ~i~ có một chỉ số năng lực đúng bằng ~a_i~. Tổ chức nhân sự của công ty XYZ có dạng đồ thị cây. Mỗi nhân viên có đúng một cấp trên trực tiếp, có một nhân viên duy nhất là tổng giám đốc, không ai là cấp trên của nhân viên này. Ta gọi nhân viên ~u~ là cấp trên của nhân viên ~v~ nếu nhân viên ~u~ là cấp trên trực tiếp của nhân viên ~v~, hoặc nhân viên ~u~ là cấp trên của nhân viên ~w~ và nhân viên ~w~ là cấp trên trực tiếp của nhân viên ~v~.
Hai nhân viên ~i~ và ~j~ là một cặp đôi hoàn hảo nếu họ thỏa mãn hai điều kiện sau:
- ~i~ là cấp trên của ~j~.
- Chênh lệch năng lực giữa hai nhân viên không vượt quá ~k~, tức là ~|a_i - a_j| \le k~.
Yêu cầu: Hãy giúp ban lãnh đạo công ty XYZ đếm số cặp đôi hoàn hảo này.
Input
- Dòng đầu tiên chứa 2 số nguyên ~n, k~ (~1 \le n, k \le 10^5~).
- Dòng tiếp theo chứa ~n~ số nguyên ~a_1, a_2, \dots, a_n~.
- ~n - 1~ dòng tiếp theo, mỗi dòng chứa 2 số nguyên ~u, v~ (~1 \le u, v \le n~) miêu tả mối quan hệ nhân viên ~u~ là cấp trên trực tiếp của nhân viên ~v~. Dữ liệu đảm bảo các mối quan hệ trong công ty tạo thành một cấu trúc cây.
Output
- Ghi ra một dòng duy nhất là kết quả của bài toán.
Sample Input 1
5 2
3 2 3 1 1
1 4
1 5
3 2
3 1
Sample Output 1
4
Bình luận