Top-K Sampling

Unlock premium problems

This is a Premium problem. Upgrade to unlock the full editor, test cases, and solution.