美文网首页
两个数组的交集

两个数组的交集

作者: 极客匠 | 来源:发表于2019-11-20 01:39 被阅读0次

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

示例 1:

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

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

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

解题思路

1、遍历两个数组,如果元素都出现相同时,则将它添加到list中,并让nums2删除该元素,直到循环结束,就可得出两个数组的交集数组。

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
            list_intersect = []
            for i in nums1:
                for j in nums2:
                    if(i == j):
                        list_intersect.append(i)
                        nums2.remove(j)
                        break
            return list_intersect
        
        

2、采用双指针将两个列表中的共同元素抠下来。

  1. 首先对两个数组进行排序
  2. 当遇到不同元素时,数值小的那个列表的指针向前移动
  3. 直到循环结束
class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
            list = []
            nums1.sort()
            nums2.sort()
            i = j =0
            while i < len(nums1) and j < len(nums2):
                if(nums1[i] == nums2[j]):
                    list.append(nums1[i])
                    i += 1
                    j += 1
                elif (nums1[i] < nums2[j]):
                    i += 1
                else :
                    j += 1
            return list
        
        

相关文章

网友评论

      本文标题:两个数组的交集

      本文链接:https://www.haomeiwen.com/subject/cojnictx.html