题解 | #多数组第 K 小数#
多数组第 K 小数
https://www.nowcoder.com/practice/41796daa4c7e4e5ab984b2c16c24a1de
# 两个数组用双指针,多个数组用堆 c1, c2 = 0, 0 num1, num2 = 0, 0 for _ in range(target): num1 = arr1[c1] if c1 < len(arr1) else float('inf') num2 = arr2[c2] if c2 < len(arr2) else float('inf') if num1 > num2: c2 += 1 else: c1 += 1 return num1 if num1 < num2 else num2 # 用堆,全部放入然后取出 # import heapq # q = [] # res = 0 # for i in range(len(arr1)): # heapq.heappush(q, arr1[i]) # for i in range(len(arr2)): # heapq.heappush(q, arr2[i]) # for i in range(target): # res = heapq.heappop(q) # return res