Tô màu tuyệt vời - 2
Đề bài
Mô tả
Cho một dãy số nguyên và màu. Ta muốn tô màu dãy này. Một cách tô được gọi là đẹp nếu thỏa mãn đồng thời các điều kiện sau:
- Mỗi phần tử của dãy hoặc được tô bằng đúng một trong màu, hoặc không được tô.
- Hai phần tử bất kỳ được tô cùng một màu thì phải có giá trị khác nhau (tức là không có hai phần tử bằng nhau cùng màu).
- Với mỗi màu, gọi số phần tử được tô màu đó là một số — tất cả số này phải bằng nhau.
- Tổng số phần tử được tô là lớn nhất trong tất cả các cách tô thỏa mãn ba điều kiện trên.
Hãy tìm một cách tô màu đẹp cho dãy đã cho.
Dữ liệu vào
- Dòng đầu chứa số nguyên — số lượng bộ dữ liệu.
- Với mỗi bộ dữ liệu:
- Dòng thứ nhất chứa hai số nguyên và — độ dài dãy và số màu.
- Dòng thứ hai chứa số nguyên .
Dữ liệu ra
Với mỗi bộ dữ liệu, in ra một dòng gồm số nguyên () cách nhau bởi dấu cách, trong đó:
- nếu phần tử thứ không được tô;
- nếu phần tử thứ được tô bằng màu .
Nếu có nhiều đáp án, in ra một đáp án bất kỳ.
Ràng buộc
- Tổng trên tất cả các bộ dữ liệu không vượt quá .
Ví dụ
| Input | Output | Giải thích |
|---|---|---|
| 3 10 3 3 1 1 1 1 10 3 10 10 2 4 4 1 1 1 1 1 1 1 |
1 3 1 2 0 3 2 1 2 3 1 2 3 4 1 |
Bộ 1: phần tử được tô, mỗi màu phần tử; một phần tử (giá trị thứ tư) bị bỏ. Bộ 2: bốn số tô bốn màu khác nhau. Bộ 3: một phần tử tô màu . |
| 2 13 1 3 1 4 1 5 9 2 6 5 3 5 8 9 13 3 3 1 4 1 5 9 2 6 5 3 5 8 9 |
1 1 1 0 1 1 1 1 0 0 0 1 0 1 3 2 1 3 3 2 3 1 2 2 0 1 |
Bộ 1 (): mỗi giá trị chỉ được tô một lần, nên với các giá trị lặp lại (1, 5, 3, 9) chỉ giữ một bản. Bộ 2 (): tô được phần tử, mỗi màu phần tử. |
Trong cả hai ví dụ trên, đáp án không duy nhất — bất kỳ cách tô đẹp nào cũng được chấp nhận.
Bình luận