Trung tâm mua sắm (DHBB 2021)

Xem PDF

Điểm: 400 (p) Thời gian: 1.0s Bộ nhớ: 256M Input: bàn phím Output: màn hình

Một trung tâm mua sắm có \(n\) kiốt (kiosk) đánh số từ \(1\) tới \(n\), kiốt thứ \(i\) bán mặt hàng \(c_i\)
. Siêu thị
\(n − 1\) con đường hai chiều đánh số từ \(1\) tới \(n − 1\), con đường thứ \(i\) nối giữa kiốt \(u_i\) và kiốt \(v_i\)
.
Hệ thống các con đường đi đảm bảo sự đi lại giữa hai kiốt bất kỳ.

Trong thời kỳ dịch bệnh, người ta muốn ngưng hoạt động một số kiốt để dễ dàng kiểm soát các
hoạt động mua bán cũng như giao tiếp với khách hàng. Khi một kiốt bị ngừng hoạt động, tất cả
các con đường nối tới kiốt đó đều bị chặn để đảm bảo an ninh. Ngoài ra vì không muốn ảnh
hưởng nhiều tới khách hàng, siêu thị muốn lập phương án sao cho các kiốt vẫn còn hoạt động
phải thỏa mãn hai điều kiện sau:

  • Các kiốt còn hoạt động phải liên thông với nhau: Tức là giữa hai kiốt bất kỳ vẫn được mở
    cửa phải tồn tại đường đi (qua các con đường không bị chặn)
  • Tất cả các mặt hàng mang số hiệu từ \(1\) tới \(k\) (là những mặt hàng thiết yếu) đều phải có bán
    ở ít nhất một kiốt còn hoạt động.

Hai phương án được gọi là khác nhau nếu có một kiốt bị ngưng hoạt động trong một phương án
nhưng được phép hoạt động trong phương án còn lại.
Yêu cầu: Hãy cho biết có bao nhiêu phương án khác nhau thỏa mãn điều kiện nêu trên.

Input

Vào từ file văn bản KIOSKS.INP

  • Dòng 1 chứa số nguyên dương \(n \le 10^4 , k \le 10\)
  • Dòng 2 chứa n số nguyên dương \(c_1, c_2, ... , c_n (\forall i: c_i \le n)\)
  • \(n − 1\) dòng tiếp theo, dòng thứ \(i\) chứa hai số nguyên dương \(u_i , v_i\)

Output

  • Ghi ra file văn bản KIOSKS.OUT một số nguyên duy nhất là số dư trong phép chia: số
    phương án thỏa mãn điều kiện đề bài cho \(1000000007 (10^9 + 7)\)

Scoring

  • Subtask \(1\) (\(30\%\) số điểm): \(k = 1\)
  • Subtask \(2\) (\(30\%\) số điểm): Mỗi kiốt có không quá 2 con đường nối tới nó.
  • Subtask \(3\) (\(40\%\) số điểm): Không có ràng buộc bổ sung ngoài các ràng buộc đã nêu trong đề bài.

Example

Test 1

Input
4 3
1 2 4 3
1 2
2 3
3 4
Output
1

Test 2

Input
5 2
1 2 2 2 3
1 2
1 3
1 4
2 5
Output
11

Bình luận

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