至少要挑选 K 双相同颜色的袜子

程序员们经常会遇到需要对一组数据进行分类的问题。若这组数据是有颜色属性的袜子，那么我们就需要找出至少 K 双相同颜色的袜子。

解决方案

我们可以使用哈希表来记录每种颜色的袜子的数量，然后遍历哈希表，找出数量大于等于 K 的颜色即可。

def find_k_pairs(socks, k):
    counts = {}
    for sock in socks:
        if sock in counts:
            counts[sock] += 1
        else:
            counts[sock] = 1

    for sock, count in counts.items():
        if count >= k:
            return sock

    return None

时间复杂度为 O(N)，其中 N 是袜子的数量。

示例

假设有如下袜子：

socks = ["red", "blue", "green", "red", "yellow", "red", "green"]

我们可以调用 find_k_pairs(socks, 3) 来找出至少 3 双相同颜色的袜子，输出结果如下：

"red"

因为红色袜子有 3 只，符合条件。

总结

使用哈希表可以很方便地解决分类问题，时间复杂度也比较优秀。同时，在使用哈希表时要注意处理好哈希冲突和边界情况。