两个数组的交集 II(Java)

两个数组的交集II

题目:

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

思想:

思想一:

使用哈希表记录数组1中的数和其对应频次。遍历数组2,如果当前索引的数在哈希表中存在,则把它丢进结果list中,并把它在哈希表中的频次-1,如果频次为0就移除它。完成遍历后,list中就是我们希望得到的结果。

思想二:

首先对两个数组进行排序,然后就是双指针,p1指向nums1,p2指向nums2: 如果nums1[p1] == nums2[p2],说明俩数组中都有这个数,是其交集,所以将它丢入list中。如果不等,则移动小的那个指针。

代码:

代码一:

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        HashMap<Integer, Integer> map = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        for(int num : nums1) {
            if(!map.containsKey(num)) map.put(num, 1);
            else map.put(num, map.get(num) + 1);
        }
        for(int num : nums2) {
            if(map.containsKey(num)) {
                map.put(num, map.get(num) - 1);
                if(map.get(num) == 0) map.remove(num);
                list.add(num);
            }
        }
        int[] res = new int[list.size()];
        for(int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }
}

代码二:

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        List<Integer> list = new ArrayList<>();
        int p1 = 0, p2 = 0;
        while(p1 < nums1.length && p2 < nums2.length) {
            if(nums1[p1] < nums2[p2]) p1++;
            else if(nums1[p1] > nums2[p2]) p2++;
            else {
                list.add(nums1[p1]);
                p1++;
                p2++;
            }
        }
        int[] res = new int[list.size()];
        for(int i = 0; i < res.length; i++) res[i] = list.get(i);
        return res;
    }
}
发布了36 篇原创文章 · 获赞 2 · 访问量 977

猜你喜欢

转载自blog.csdn.net/y18771025420/article/details/102703982