输入一个序列，代码实现束搜索k=2

以下是一个简单的 Python 代码，用于实现束搜索（beam search）算法，其中 k=2：

import heapq

def beam_search(sequence, k=2):
    # 初始化堆和初始序列
    heap = [(0, [sequence[0]])]
    # 对于序列中剩余的元素，进行 beam search
    for i in range(1, len(sequence)):
        candidates = []
        # 对于每个前缀序列，扩展后缀元素并计算分数
        for score, prefix in heap:
            for j in range(len(prefix)):
                candidate = prefix[:j] + [sequence[i]] + prefix[j+1:]
                # 计算分数，分数越低越好
                candidate_score = -sum(candidate)
                candidates.append((candidate_score, candidate))
        # 选择分数最佳的 k 个序列
        heap = heapq.nsmallest(k, candidates)
    # 返回最佳序列
    return heap[0][1]

这个算法的时间复杂度为 O(kn^2logk)，其中 n 是序列的长度。