Cây của Appleman
Đề bài
Mô tả
Cho một cây có đỉnh được đánh số từ đến . Mỗi đỉnh được tô màu đen hoặc trắng, và đảm bảo có ít nhất một đỉnh đen.
Xét một tập gồm cạnh của cây (). Khi xóa cạnh này khỏi cây, cây sẽ bị tách thành thành phần liên thông (mỗi thành phần cũng là một cây con với các đỉnh đã được tô màu).
Hãy đếm số tập cạnh sao cho sau khi xóa, mỗi thành phần liên thông chứa đúng một đỉnh đen. In kết quả theo modulo .
Dữ liệu vào
- Dòng đầu chứa số nguyên — số đỉnh của cây.
- Dòng thứ hai chứa số nguyên (), trong đó nghĩa là có một cạnh nối đỉnh với đỉnh .
- Dòng thứ ba chứa số nguyên (). Nếu thì đỉnh màu đen, ngược lại màu trắng.
Dữ liệu ra
Một số nguyên duy nhất — số tập cạnh hợp lệ theo modulo .
Ràng buộc
- Có ít nhất một đỉnh đen.
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 3 0 0 0 1 1 |
2 | Cây có 3 đỉnh: đỉnh 0 nối với 1 và 2. Đỉnh 0 trắng, đỉnh 1 và 2 đen. Có hai cách: (1) xóa cạnh (0,1) và (0,2) thành 3 thành phần — không hợp lệ vì đỉnh 0 không có đen; (2) xóa cạnh (0,1) → {0,2} và {1}; (3) xóa cạnh (0,2) → {0,1} và {2}. Hai cách (2) và (3) hợp lệ. |
| 6 0 1 1 0 4 1 1 0 0 1 0 |
1 | Chỉ có một cách chia hợp lệ. |
| 10 0 1 2 1 4 4 4 0 8 0 0 0 1 0 1 1 0 0 1 |
27 | Có 27 cách phân hoạch hợp lệ. |
Bình luận