Chọn ĐTQG Hà Nam 2024 - Công viên vui vẻ
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
Công viên Happy là một nơi giải trí nổi tiếng, nơi có nhiều điểm tham quan thú vị. Nằm dọc trên đường đi trong công viên có ~n~ trạm được đánh số từ ~1~ đến ~n~. Mỗi trạm chỉ chuyên phục vụ một loại trò chơi trong tổng số ~m~ loại trò chơi khác nhau.
Tuệ Minh được mẹ cho đi công viên Happy, trên tay bạn là sơ đồ công viên gồm một bảng mô tả chi tiết loại của trò chơi ở mỗi trạm. Bạn ấy rất hứng thú và muốn đánh giá trải nghiệm của mình bằng cách tính toán tổng chỉ số vui vẻ (happiness index) cho mỗi đoạn gồm một số trạm liên tiếp nhau, từ trạm thứ ~l~ đến trạm thứ ~r~.
Chỉ số vui vẻ của một đoạn các trạm từ trạm ~l~ đến ~r~ được định nghĩa như sau:
- Giả sử sau khi trải nghiệm toàn bộ các trò chơi từ trạm thứ ~l~ đến trạm thứ ~r~, có ~k~ loại trò chơi bạn chưa tham gia, lần lượt là ~c_1, c_2, \dots, c_k~.
Chỉ số vui vẻ của đoạn từ trạm ~l~ đến ~r~ được tính bằng tổng của tất cả các loại trò chơi mà bạn chưa tham gia trong đoạn này, cụ thể là ~c_1 + c_2 + \dots + c_k~.
Yêu cầu: Bạn hãy giúp Tuệ Minh tính tổng chỉ số vui vẻ cho tất cả các đoạn trạm liên tiếp dọc trên con đường đi trong công viên. Kết quả có thể rất lớn, vì vậy bạn hãy đưa ra kết quả sau khi chia lấy dư cho số ~998244353~.
Input
- Dòng đầu tiên chứa hai số nguyên ~n~ và ~m~ ~(1 \le n \le 2 \cdot 10^5, 1 \le m \le 10^9)~ lần lượt là số trạm và số loại trò chơi.
- Dòng thứ hai chứa ~n~ số nguyên ~a_1, a_2, \dots, a_n~ ~(1 \le a_i \le m)~ mô tả loại trò chơi tại từng trạm.
Output
In ra tổng chỉ số vui vẻ của tất cả các đoạn trạm liên tiếp sau khi chia lấy dư cho ~998244353~.
Scoring
| Subtask | Điểm | Ràng buộc |
|---|---|---|
| 1 | ~30\%~ | ~1 \le n, m \le 500~ |
| 2 | ~40\%~ | ~1 \le n, m \le 5000~ |
| 3 | ~30\%~ | Không có ràng buộc gì thêm |
Sample Input 1
3 4 1 2 3
Sample Output 1
40
Sample Input 2
8 4
2 3 1 2 4 3 1 3
Sample Output 2
124
Notes
Ở ví dụ 1, xét các đoạn liên tiếp:
- ~[1, 1]~ : không chơi được các loại trò chơi ~2, 3, 4~, chỉ số vui vẻ là ~9~.
- ~[1, 2]~ : không chơi được các loại trò chơi ~3, 4~, chỉ số vui vẻ là ~7~.
- ~[1, 3]~ : không chơi được các loại trò chơi ~4~, chỉ số vui vẻ là ~4~.
- ~[2, 2]~ : không chơi được các loại trò chơi ~1, 3, 4~, chỉ số vui vẻ là ~8~.
- ~[2, 3]~ : không chơi được các loại trò chơi ~1, 4~, chỉ số vui vẻ là ~5~.
- ~[3, 3]~ : không chơi được các loại trò chơi ~1, 2, 4~, chỉ số vui vẻ là ~7~.
Vậy tổng chỉ số vui vẻ là ~40~.
Bình luận