【周赛】【简单】6362. 合并两个二维数组 - 求和法

Lan
Lan
2023-02-19 / 0 评论 / 93 阅读 / 正在检测是否收录...
温馨提示:
本文最后更新于2023年02月19日,已超过641天没有更新,若内容或图片失效,请留言反馈。

www.lanol.cn

题目

给你两个 二维 整数数组 nums1 和 nums2.

nums1[i] = [idi, vali] 表示编号为 idi 的数字对应的值等于 vali 。
nums2[i] = [idi, vali] 表示编号为 idi 的数字对应的值等于 vali 。
每个数组都包含 互不相同 的 id ,并按 id 以 递增 顺序排列。

请你将两个数组合并为一个按 id 以递增顺序排列的数组,并符合下述条件:

只有在两个数组中至少出现过一次的 id 才能包含在结果数组内。
每个 id 在结果数组中 只能出现一次 ,并且其对应的值等于两个数组中该 id 所对应的值求和。如果某个数组中不存在该 id ,则认为其对应的值等于 0 。
返回结果数组。返回的数组需要按 id 以递增顺序排列。

 

示例 1:
输入:nums1 = [[1,2],[2,3],[4,5]], nums2 = [[1,4],[3,2],[4,1]]
输出:[[1,6],[2,3],[3,2],[4,6]]
解释:结果数组中包含以下元素:

  • id = 1 ,对应的值等于 2 + 4 = 6 。
  • id = 2 ,对应的值等于 3 。
  • id = 3 ,对应的值等于 2 。
  • id = 4 ,对应的值等于5 + 1 = 6 。
    示例 2:
    输入:nums1 = [[2,4],[3,6],[5,5]], nums2 = [[1,3],[4,3]]
    输出:[[1,3],[2,4],[3,6],[4,3],[5,5]]
    解释:不存在共同 id ,在结果数组中只需要包含每个 id 和其对应的值。
    提示:
  • <= nums1.length, nums2.length <= 200
    nums1[i].length == nums2[j].length == 2
  • <= idi, vali <= 1000
    数组中的 id 互不相同
    数据均按 id 以严格递增顺序排列

解法

今天早上刚好看见有个周赛,于是参与了一下,我还是太菜了,只做了一个半。

class Solution:
    def mergeArrays(self, nums1: List[List[int]], nums2: List[List[int]]) -> List[List[int]]:
        fast, slow = 0, 0
        while fast < len(nums1) and slow < len(nums2):
            if nums1[fast][0] < nums2[slow][0]:
                fast += 1
            elif nums1[fast][0] > nums2[slow][0]:
                nums1.insert(fast, nums2[slow])
                slow += 1
            else:
                nums1[fast][1] += nums2[slow][1]
                slow += 1
        return nums1 + nums2[slow:] if slow < len(nums2) else nums1
0

评论 (0)

取消