两个数组的交集问题java实现

1.两个数组的交集

https://leetcode-cn.com/problems/intersection-of-two-arrays/

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

 

示例 1：

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

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

说明：

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

思路：双指针遍历，用HashSet存储结果使其不重复

 

import java.util.*;
class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);Arrays.sort(nums2);
        //用HashSet存储不重复的结果
        Set<Integer> set=new HashSet<Integer>();
        int i=0,j=0;
        while(i<nums1.length&&j<nums2.length){
            if(nums1[i]==nums2[j]){
                set.add(nums1[i]);
                i++;
                j++;
            }
            else if(nums1[i]<nums2[j]){
                i++;
            }
            else{
                j++;
            }
        }
        int[] res=new int[set.size()];
        int k=0;
        //HashSet用增强型for循环遍历，结果赋给数组
        for(int temp:set){
            res[k]=temp;
            k++;
        }
        return res;
    }
}

2.两个数组的交集 II

https://leetcode-cn.com/problems/intersection-of-two-arrays-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]
说明：

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

思路：双指针遍历，用ArrayList存储可重复的结果

import java.util.*;
class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        //用链表存放交集数据
        List<Integer> list=new ArrayList<Integer>();
        //对数组进行排序
        Arrays.sort(nums1);Arrays.sort(nums2);
        //双指针查找
        int i=0,j=0;
        while(i<nums1.length&&j<nums2.length){
            if(nums1[i]==nums2[j]){
                list.add(nums1[i]);
                i++;
                j++;
            }
            else if(nums1[i]<nums2[j]){
                i++;
            }
            else{
                j++;
            }
        }
        int[] res=new int[list.size()];
        //将链表数据赋给数组
        for(int k=0;k<res.length;k++){
            res[k]=list.remove(0);
        }
        return res;
    }
}