349. 两个数组的交集

题目

给定两个数组，编写一个函数来计算它们的交集。
示例 1：
输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2：
输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

思路

排序+双数组

代码

python版

class Solution:
    def intersection(self, nums1: List[int], nums2: List[int]) -> List[int]:
        ans = []
        i,j=0,0
        nums1.sort()
        nums2.sort()
        while i<len(nums1) and j<len(nums2):
            if nums1[i]==nums2[j]:
                if i==0 or nums1[i]!=nums1[i-1]: 
                	ans.append(nums1[i])
                i+=1
                j+=1
            elif nums1[i]<nums2[j]:
                i+=1
            else:
                j+=1
        return ans 

Java版

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        ArrayList<Integer> list = new ArrayList<Integer>();
        int len1 = nums1.length,len2 = nums2.length;
        int i=0,j=0;
        while(i<len1&&j<len2){
            if(nums1[i]==nums2[j]){
                if(i==0||nums1[i]!=nums1[i-1]){
                    list.add(nums1[i]);
                }
                i++;
                j++;
            }
            else if(nums1[i]<nums2[j]){
                i++;
            }
            else{
                j++;
            }
        }
        int [] ans = new int[list.size()];
        int index = 0;
        for(int num:list){
            ans[index++]=num;
        }
        return ans;

    }
}

复杂度

时间复杂度：O(mlogm+nlogn)
空间复杂度：O(logm+logn)，其中 m 和 n 分别是两个数组的长度。空间复杂度主要取决于排序使用的额外空间。