Leetcode—— 350. 两个数组的交集 II

350. 两个数组的交集 II

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

示例 1:

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路:

  • 对数组 nums1 和 nums2 排序。
  • 初始化指针 i,j 为 0。
  • 指针 i 指向 nums1,指针 j 指向 nums2:
  • 如果 nums1[i] < nums2[j],则 i++。
  • 如果 nums1[i] > nums2[j],则 j++。
  • 如果 nums1[i] == nums2[j],将元素拷贝到新列表里,且 i++,j++。

程序代码(Python):

class Solution:
    def intersect(self, nums1, nums2):
        nums1.sort()
        nums2.sort()
        len1 = len(nums1)
        len2 = len(nums2)
        l = []
        i = j = 0
        while(i nums1[i]:
                i += 1
            elif nums2[j] < nums1[i]:
                j += 1
            else:
                l.append(nums1[i])
                i += 1
                j += 1
        return l

nums1 = [1,2,2,1]
nums2 = [2,2]
s = Solution()
print(s.intersect(nums1, nums2))

程序代码(C++):

class Solution {
public:
    vector intersect(vector& nums1, vector& nums2) 
{
        vector res;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int i=0,j=0;
        while(inums2[j])
                j++;
            else
                res.push_back(nums1[i]), i++, j++;
        }
        return res;
    }
};

 

你可能感兴趣的:(Python,C++,LeetCode)