Chia ba
Đề bài
Mô tả
Cho một dãy gồm số nguyên không âm . Hãy chia dãy thành ba đoạn con, đoạn thứ nhất gồm một hoặc một số các phần tử đầu tiên của dãy , đoạn thứ hai gồm một hoặc một số phần tử tiếp theo của dãy , đoạn thứ ba gồm các phần tử còn lại của dãy thỏa mãn các điều kiện sau:
- Tổng các phần tử của đoạn con thứ nhất chia hết cho .
- Tổng các phần tử của đoạn con thứ hai chia dư .
- Tổng các phần tử của đoạn con thứ ba chia dư .
Hai cách chia được coi là khác nhau nếu vị trí kết thúc của đoạn thứ nhất hoặc của đoạn thứ hai khác nhau.
Yêu cầu: Hãy cho biết có bao nhiêu cách chia dãy thỏa mãn các điều kiện trên.
Dữ liệu vào
- Dòng đầu tiên chứa số nguyên dương là số phần tử của dãy .
- Dòng thứ hai chứa số nguyên với .
Dữ liệu ra
In ra một số nguyên duy nhất là số cách chia dãy thỏa mãn các điều kiện trên.
Ràng buộc
- .
- với .
Subtasks
Điểm được tính theo từng test.
- Subtask 1 ( số test): .
- Subtask 2 ( số test): .
- Subtask 3 ( số test): .
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 7 2 1 3 0 4 0 5 |
6 | Đoạn thứ nhất là , hoặc (tổng , , ); đoạn thứ hai kết thúc tại hoặc (tổng tiền tố chia dư ), đoạn thứ ba là hoặc (tổng chia dư ). Có cách. |
| 5 0 0 1 1 1 |
2 | Ví dụ bổ sung (không có trong đề gốc). Hai cách: và . Đoạn thứ nhất không được rỗng nên không tính cách . |
Bình luận