Leetcode349：两个数组的交集

题目

给定两个数组，编写一个函数来计算它们的交集。
示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2：

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

说明：

• 输出结果中的每个元素一定是唯一的。
• 我们可以不考虑输出结果的顺序。

思路一

迭代检查第一个数组的元素是否在第二个数组中，如果存在则将其添加到结果中，否则继续。

import java.util.HashSet;
import java.util.Set;
import java.util.Scanner;
import  java.util.Arrays;
public class IntersectionI {
    public int[] intersection(int[] nums1, int[] nums2) {
        if(nums1.length == 0 || nums2.length == 0) return new int[0];
        if(nums1.length > nums2.length) intersection(nums2, nums1);
        Set<Integer> list = new HashSet<>();
        Set<Integer> set = new HashSet<>();
        //利用set集合的特性，不能存储相同元素
        for(int num : nums2){
            set.add(num);
        }
        
        for(int num : nums1){
            if(set.contains(num)) list.add(num);
        }
        int[] ret = new int[list.size()];
        int i = 0;
        for(int num : list) ret[i++] = num;
        return ret;
    }

    public static void main(String[] args){
        String[] str = new String[0];
        Scanner  sc = new Scanner(System.in);
        str = sc.nextLine().split(",");//这对输入数据的格式有要求
        int[] nums1 = new int[str.length];
        for(int i = 0; i < str.length; i++) {
            nums1[i] = Integer.parseInt(str[i]);
        }

        String[] str2 = new String[0];
        str2 = sc.nextLine().split(",");
        int[] nums2 = new int[str2.length];
        for(int i = 0; i < str2.length; i++) {
            nums2[i] = Integer.parseInt(str2[i]);
        }

        IntersectionI test = new IntersectionI();
        int[] ret = test.intersection(nums1, nums2);
        System.out.println(Arrays.toString(ret));
    }
}

思路二

先排序，然后利用双指针，一个指向nums1,一个指向nums2:

• nums[i] > nums[j]: j++
• nums[i]<nums[j]:i++
• nums[i] = nums[j]:在判断这个值是否之前已经添加到结果中，如果存在，则继续，否则添加。

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        if(nums1.length == 0 || nums2.length == 0) return new int[0];
        if(nums1.length > nums2.length) intersection(nums2, nums1);
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        List<Integer> list = new ArrayList<>();
        for(int i = 0, j = 0; i < nums1.length && j < nums2.length;){
            if(nums1[i] > nums2[j]){
                j++;
            }else if(nums1[i] < nums2[j]){
                i++;
            }else{
                if(!list.contains(nums1[i]))
                    list.add(nums1[i]);
                i++;j++;
            }
        }
        int[] ret = new int[list.size()];
        for(int i = 0; i < list.size(); i++){
            ret[i] = list.get(i);
        }
        return ret;
    }
}