LeetCode-350(两个数组的交集 II)

v1
  • 双遍历 + 字典 遍历两个数组其中元素出现的次数,在用字典存储
    时间O(2n) + O(len(set())) 空间 O(2n)
class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        dst = list(set(nums1) & set(nums2))

        ret_map = {}

        for i in nums1:
            if i in dst:
                if ret_map.get(i, None):
                    ret_map[i] += 1
                else:
                    ret_map[i] = 1
        
        ret_map2 = {}
        for j in nums2:
            if j in dst:
                if ret_map2.get(j, None):
                    ret_map2[j] += 1
                else:
                    ret_map2[j] = 1

        ret_lst = []
        for r in dst:
            if ret_map2[r] < ret_map[r]:
                ret_lst.extend([r] * ret_map2[r])
            
            else:
                ret_lst.extend([r] * ret_map[r])

        return ret_lst
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容