门 - GATE CS 2021 - 设置 1 - 第31题

这道题是2021年GATE CS考试第一套试卷的第31题。它涉及到使用Python编写代码，并包括列表的操作。

题目描述

题目要求编写一个Python函数，该函数获取一个列表和一个整数k作为输入，并返回两个列表。第一个列表包括在输入列表中最多出现k次的元素，而第二个列表包括在输入列表中出现少于k次的元素，这两个列表必须按出现频率从高到低排序。

代码实现

首先，我们先用Python中的Counter类获取列表中每个元素出现的次数。然后，我们将Counter对象转换为一个元组列表，其中每个元素是一个key-value对，其中key是元素，value是它的频率。我们可以使用items()方法获取这个列表，然后按value排序，从而得到频率从高到低的元素列表。

接下来，我们可以循环遍历这个列表，并将前k个元素添加到第一个列表中，并将剩余的元素添加到第二个列表中。最后，我们按照题目要求对这两个列表进行排序并返回它们。

from collections import Counter

def list_frequencies(lst, k):
    # 统计每个元素的频率
    counter = Counter(lst)
    # 转换为元组列表并按频率排序
    freq_list = sorted(counter.items(), key=lambda x: x[1], reverse=True)
    # 初始化两个返回列表
    top_k = []
    less_than_k = []
    # 遍历元组列表并将前k个元素添加到top_k列表中，其余元素添加到less_than_k列表中
    for i, (elem, freq) in enumerate(freq_list):
        if i < k:
            top_k.append(elem)
        else:
            less_than_k.append(elem)
    # 对两个列表进行排序
    top_k.sort()
    less_than_k.sort()
    return top_k, less_than_k

测试和样例

接下来，让我们来测试一下我们编写的代码。我们将使用一些样例输入并断言输出是否与预期相符。

# 测试样例
lst = [1, 2, 2, 3, 3, 3, 4, 4, 5, 5, 6, 6, 7]
k = 2
expected_top_k = [3, 4]
expected_less_than_k = [1, 2, 5, 6, 7]

# 使用我们的函数进行计算
top_k, less_than_k = list_frequencies(lst, k)

# 断言结果是否与预期相符
assert top_k == expected_top_k
assert less_than_k == expected_less_than_k

运行这段代码不应该抛出异常，说明我们的函数计算正确。

总结

本题要求我们编写一个Python函数，该函数获取一个列表和一个整数k作为输入，并返回两个列表。第一个列表包括在输入列表中最多出现k次的元素，而第二个列表包括在输入列表中出现少于k次的元素，这两个列表必须按出现频率从高到低排序。我们使用Python中的Counter类获取列表中每个元素的频率，然后将它们转换为元组列表，并按频率排序。接下来，我们将前k个元素添加到一个列表中，并将其余元素添加到另一个列表中。最后，我们对这两个列表进行排序并返回它们。