题解 | #多数组第 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 


全部评论

相关推荐

牛客316659795号:不是,证明hr初筛已经过了,要投给部门筛一遍
点赞 评论 收藏
分享
xwqlikepsl:感觉很厉害啊,慢慢找
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务