两个数组的交集 II

两个数组的交集 II

题目链接

解题思路

先排序，然后双指针遍历

c++解法

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        //排序
        sort(nums1.begin(),nums1.end());
        sort(nums2.begin(),nums2.end());
        vector<int> ans;
        int i=0,j=0;
        while(i<nums1.size()&&j<nums2.size()){
            if(nums1[i]==nums2[j]){
                ans.push_back(nums1[i]);
                i++;
                j++;
            }else if(nums1[i]>nums2[j]){
                j++;
            }else if(nums1[i]<nums2[j]){
                i++;
            }
        }
        return ans;
    } 
};

java解法

java的排序函数为Arrays.sort(nums1);

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        //排序
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        ArrayList<Integer> arr=new ArrayList<Integer>();
        //遍历
        int i=0,j=0;
        while(i<nums1.length&&j<nums2.length){
            if(nums1[i]==nums2[j]){
                arr.add(nums1[i]);
                i++;
                j++;
            }else if(nums1[i]>nums2[j]) j++;
            else if(nums1[i]<nums2[j]) i++;
        }
        //创建返回数组
        int[] ans=new int[arr.size()];
        for(int k=0;k<arr.size();k++) ans[k]=arr.get(k);
        return ans;
    }
}

python解法

class Solution:
    def intersect(self, nums1: List[int], nums2: List[int]) -> List[int]:
        #排序
        nums1.sort()
        nums2.sort()
        ans=[]
        #遍历
        i=0
        j=0
        while i<len(nums1) and j<len(nums2):
            if nums1[i]==nums2[j]:
                ans.append(nums1[i])
                i+=1
                j+=1
            elif nums1[i]<nums2[j]: i+=1
            elif nums1[i]>nums2[j]: j+=1
        return ans