Tô màu sỏi
Đề bài
Mô tả
Trên bàn có đống sỏi, đống thứ chứa viên sỏi. Bạn cần tô mỗi viên sỏi bằng một trong màu cho trước sao cho với mọi màu và mọi cặp đống , , số viên sỏi màu ở hai đống chênh nhau không quá .
Nói cách khác, gọi là số viên sỏi màu trong đống thứ . Khi đó với mọi và mọi phải có:
Không bắt buộc phải dùng hết màu: nếu màu không xuất hiện trong đống thì .
Hãy chỉ ra một cách tô thỏa mãn, hoặc cho biết không tồn tại cách tô nào.
Dữ liệu vào
- Dòng đầu chứa hai số nguyên dương và : số đống sỏi và số màu.
- Dòng thứ hai chứa số nguyên dương : số viên sỏi trong từng đống.
Dữ liệu ra
Nếu không tồn tại cách tô thỏa mãn, in ra một dòng chứa NO.
Ngược lại, dòng đầu in ra YES, sau đó in dòng: dòng thứ chứa số nguyên, số thứ là màu của viên sỏi thứ trong đống (mỗi màu là một số nguyên trong đoạn ). Nếu có nhiều đáp án, in ra đáp án bất kỳ.
Ràng buộc
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 4 4 1 2 3 4 |
YES 1 1 2 1 2 3 1 2 3 4 |
Mỗi màu xuất hiện hoặc lần ở mỗi đống nên mọi hiệu đều không vượt quá . Đáp án khác cũng được chấp nhận. |
| 5 2 3 2 4 1 3 |
NO | Đống lớn nhất có viên, đống nhỏ nhất có viên. Với chỉ màu, đống viên phải có một màu dùng ít nhất lần, trong khi đống viên dùng màu đó nhiều nhất lần. |
| 5 4 3 2 4 3 5 |
YES 1 2 3 1 2 1 2 3 4 1 2 3 1 2 3 4 1 |
Màu xuất hiện lần ở năm đống, các màu còn lại xuất hiện nhiều nhất lần mỗi đống. |
Bình luận