LeetCode:两个数组的交集 II(Python版本)

LeetCode刷题日记

    • 两个数组的交集 II
      • Python代码

两个数组的交集 II

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/

给定两个数组,编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]

说明:

输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。
进阶:

如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

Python代码

from collections import Counter
class Solution(object):
    def intersect(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        ls = []
        c1 = Counter(nums1).most_common()
        c2 = Counter(nums2).most_common()
        if len(c1) <= len(c2):
            for i in range(len(c1)):
                for j in range(len(c2)):
                    if c1[i][0] == c2[j][0]:
                        if c1[i][1] <= c2[j][1]:
                            for a in range(c1[i][1]):
                                ls.append(c1[i][0])
                        else:
                            for a in range(c2[j][1]):
                                ls.append(c2[j][0])
        else:
            for j in range(len(c2)):
                for i in range(len(c1)):
                    if c2[j][0] == c1[i][0]:
                        if c2[j][1] <= c1[i][1]:
                            for a in range(c2[j][1]):
                                ls.append(c2[j][0])
                        else:
                            for a in range(c1[i][1]):
                                ls.append(c1[i][0])
        return ls

比较麻烦,但是思路还是很清晰的,进阶问题暂时没有考虑,但是这个代码肯定是不行的了。

你可能感兴趣的:(数组,初级算法,Python)