本文共 1251 字，大约阅读时间需要 4 分钟。

为了解决这个问题，我们需要找到两个长度为N的序列A和B中各取一个数得到的N²个和中的最小的N个，并按升序输出。

方法思路

我们可以使用优先队列（最小堆）来有效地维护这些和，确保我们只保留最小的N个和。具体步骤如下：

这种方法的时间复杂度为O(N log N)，因为排序和堆操作都是O(N log N)的复杂度。

解决代码

import heapqn = int(input())A = list(map(int, input().split()))B = list(map(int, input().split()))A.sort()B.sort()heap = []i = j = 0while i < n or j < n:    sum_val = A[i] + B[j]    heapq.heappush(heap, sum_val)    if len(heap) > n:        heapq.heappop(heap)    # 确保下一个sum不小于当前sum    if i < n - 1 and j < n - 1:        if A[i + 1] + B[j] < sum_val:            i += 1        elif A[i] + B[j + 1] < sum_val:            j += 1        else:            i += 1            j += 1    else:        if i < n - 1:            i += 1        elif j < n - 1:            j += 1result = []while heap:    result.append(str(heapq.heappop(heap)))print(' '.join(result))

代码解释

这种方法确保了在处理大量数据时的效率，并且能够正确地找到最小的N个和。

转载地址：http://usrfk.baihongyu.com/