给定 logits 或概率，实现 Top-K 截断、重新归一化与随机采样。

手撕：实现一个最简单的 top - k 采样（给定 logits ／概率，取 top - k 后重新归一化采样），并说明边界情况怎么处理。