Software (OLP 30/4 2023)

Xem PDF

Nộp bài

Điểm: 1400 (thành phần)
Thời gian: 1.0s
Bộ nhớ: 256M
Input: SOFTWARE.inp
Output: SOFTWARE.out

Tác giả:
Dạng bài

Tâm rất yêu thích lập trình tạo phần mềm. Vào dịp rảnh rỗi Tâm đã thiết kế một phần mềm đơn giản. Màn hình phần mềm gồm N địa điểm (đánh số từ 1 đến N), trong đó mỗi địa điểm có đặt một bóng đèn ở trạng thái sáng hoặc tắt. Có N-1 con đường một chiều nối trực tiếp giữa các cặp điểm. Mỗi lần Tâm chạm tay vào một địa điểm X_i bất kỳ trên màn hình thì sẽ có một robot xuất phát từ địa điểm X_i di chuyển theo các con đường một chiều, cuối cùng kết thúc ở địa điểm 1. Robot không thay đổi trạng thái đèn ở địa điểm X_i và địa điểm 1, các địa điểm còn lại robot đã đi qua thì đèn ở địa điểm đó sẽ thay đổi sang trạng thái ngược lại (sáng thành tắt, tắt thành sáng)

Yêu cầu: Hãy cho biết khi Tâm thực hiện K lần chạm tay (mỗi lần chạm tay vào một địa điểm) thì sau đó sẽ có bao nhiêu địa điểm có đèn sáng. Biết rằng robot xuất phát từ địa điểm bất kỳ luôn có thể di chuyển theo các con đường một chiều đến địa điểm 1

Input, Output và Subtasks

Input: (SOFTWARE.inp)
  • Dòng đầu tiên gồm 2 số nguyên dương NK lần lượt là số địa điểm, số lần chạm tay (1\le N,K\le 10^5)
  • Dòng thứ hai gồm N số nguyên cho biết trạng thái đèn ở N địa điểm, lần lượt theo thứ tự từ địa điểm 1 đến N. Trạng thái đèn tắt là 0, sáng là 1
  • Dòng thứ i trong N-1 dòng tiếp theo gồm 2 số nguyên dương A_iB_i (1\le A_i,B_i\le N) cho biết có con đường một chiều nối trực tiếp từ địa điểm A_i đến B_i
  • Dòng cuối cùng gồm K số nguyên dương, trong đó số nguyên thứ iX_i (1\le X_i\le N) cho biết địa điểm thứ i mà Tâm thực hiện chạm tay
Output: (SOFTWARE.out)
  • Số nguyên duy nhất là kết quả cần tìm
Subtasks
  • Subtask 1 (50\%): 1\le N,K\le 5000
  • Subtask 2 (50\%): Không có ràng buộc gì thêm

Sample 1

Input (SOFTWARE.inp)
5 3
1 0 0 0 0
2 1
4 2
3 2
5 4
4 5 4
Output (SOFTWARE.out)
3
Note

Tâm chạm tay 3 lần:

  • Lần 1 ở địa điểm 4: robot đi qua các địa điểm 4,2,1 và thay đổi trạng thái đèn ở địa điểm 2
    Kết quả trạng thái 5 đèn theo thứ tự lần lượt là: 1 1 0 0 0
  • Lần 2 ở địa điểm 5: robot đi qua các địa điểm 5,4,2,1 và thay đổi trạng thái đèn ở địa điểm 4,2
    Kết quả trạng thái 5 đèn theo thứ tự lần lượt là: 1 0 0 1 0
  • Lần 3 ở địa điểm 4: robot đi qua các địa điểm 4,2,1 và thay đổi trạng thái đèn ở địa điểm 2
    Kết quả trạng thái 5 đèn theo thứ tự lần lượt là: 1 1 0 1 0

Sau 3 lần chạm tay, có 3 địa điểm có đèn sáng là địa điểm 1, 24

Sample 2

Input (SOFTWARE.inp)
4 1
0 0 0 0
3 1
2 3
4 2
4
Output (SOFTWARE.out)
2

Bình luận

Không có bình luận nào.