array_intersection_ii

  • 2022-12-14
  • 浏览 (580)

array_intersection_ii.py 源码

# 两个数组的交集II
from typing import List


class Solution:

    # 使用一个哈希表计数
    def intersect_1(self, nums1: List[int], nums2: List[int]) -> List[int]:
        m, res = {}, []
        for n in nums1:
            m[n] = m[n] + 1 if n in m else 1
        for n in nums2:
            if n in m and m[n] > 0:
                res.append(n)
                m[n] -= 1
        return res

    # 排序之后,使用双指针
    def intersect_2(self, nums1: List[int], nums2: List[int]) -> List[int]:
        nums1.sort(), nums2.sort()
        res, i, j = [], 0, 0
        while i < len(nums1) and j < len(nums2):
            if nums1[i] < nums2[j]:
                i += 1
            elif nums1[i] > nums2[j]:
                j += 1
            else:
                res.append(nums1[i])
                i, j = i + 1, j + 1
        return res

你可能感兴趣的文章

array_intersection

find_all_anagrams

group_anagram

0  赞