β

python 两个数组的交集 intersection of two arrays

Waiting For You 213 阅读

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

例子:

给定 num1 = [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]
        """
        setNums1 = set(nums1)
        setNums2 = set(nums2)
        
        result = []
        for x in setNums2:
            if x in setNums1:
                result.append(x)
        return result

用set来把list的重复元素过滤掉,然后判断是否存在,把结果保存起来
http://www.waitingfy.com/archives/3724

作者:Waiting For You
记录一些关于android,cocos2d-x,objective-c,mfc,directX,c++,数学的东西
原文地址:python 两个数组的交集 intersection of two arrays, 感谢原作者分享。