两个数组的交集Python解法

给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

例:

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

解析:

对两个数组进行去重,然后进行升序排列,从小到大依次进行比较,相同得值放入结果数组,两个数组指针有一个到顶了就退出循环比较。最后返回结果数组即可。

class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums1 = list(set(nums1))  # 去重
        nums1.sort()  # 升序排列
        nums2 = list(set(nums2))
        nums2.sort()
        i = 0  # 数组1指针
        j = 0  # 数组2指针
        res = []  # 结果数组
        while i < len(nums1) and j < len(nums2):  # 循环条件
            if nums1[i] == nums2[j]:  # 两个值想等得情况
                res.append(nums1[i])  # 将值加入结果数组
                i += 1  # 数组1指针后移
                j += 1  # 数组2指针后移
            elif nums1[i] < nums2[j]:  # 数组1得值小于数组2
                i += 1  # 数组1得指针后移
            elif nums1[i] > nums2[j]:  # 数组1得值大于数组2
                j += 1  # 数组2得指针后移
        return res  # 返回结果数组

你可能感兴趣的:(Python,开发语言,python,算法,leetcode)