Very good opportunity to apply Gosper's hack trick.

m is number of different colors in the input.

Then iterate through all subsets of size k of m.

It's really works. But I don't understand why this algo work. Can you explain me?