#2833. PARKCAT - Kefa and Park

Bộ nhớ: 256 MiB Thời gian: 1000 ms Nhập/xuất từ luồng chuẩn
Kiểu bài: Thông thường Kiểu chấm: So sánh văn bản
Đưa lên bởi: Trùm CUỐI

Đề bài

Kefa quyết định đi dạo trong công viên. Công viên này có thể được biểu diễn dưới dạng một cây có gốc tại đỉnh 1 . Trên cây có n đỉnh, một vài đỉnh trong số đó có mèo. Kefa rất sợ mèo, cậu ấy sẽ chỉ đi đến những nhà hàng nằm ở các đỉnh lá của cây.

Một đường đi từ gốc đến một đỉnh lá được gọi là "an toàn" nếu trên đường đi đó, số đỉnh có mèo xuất hiện liên tiếp không vượt quá m . Nhiệm vụ của bạn là đếm xem Kefa có thể đến được bao nhiêu nhà hàng (đỉnh lá) thông qua một đường đi "an toàn".

Dữ liệu:

  • Dòng đầu tiên chứa hai số nguyên n m .
  • Dòng thứ hai chứa n số nguyên a_1, a_2, \dots, a_n , trong đó a_i=1 nếu đỉnh i có mèo và a_i=0 nếu không.
  • n-1 dòng tiếp theo, mỗi dòng chứa hai số nguyên u v mô tả một cạnh của cây.

Kết quả: In ra một số nguyên duy nhất là số lượng đỉnh lá mà Kefa có thể đến được.

Ví dụ: Dữ liệu:

4 1
1 1 0 0
1 2
1 3
1 4

Kết quả:

2

Giới hạn:

  • 2 \le n \le 10^5
  • 1 \le m \le n
  • 1 \le u, v \le n