2020-11-02

力扣打卡题之两数组的交集

力扣题349
编写一个函数来求出来求出两个数组的交集。注意说明

思路1：可以使用hashset的方法，创建hashSet集合，然后使用暴力法进行比对（由于给定的数组中有重复元素，而Java中HashSet的特点就是不允许存储重复元素）。
代码：

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<Integer>();
        Set<Integer> set2 = new HashSet<Integer>();
        //由于HashSet不能使用索引遍历，只能使用曾强for循环
        for (int num : nums1) {
            set1.add(num);
        }
        for (int num : nums2) {
            set2.add(num);
        }
        return getIntersection(set1, set2);
    }

    public int[] getIntersection(Set<Integer> set1, Set<Integer> set2) {
        if (set1.size() > set2.size()) {
            return getIntersection(set2, set1);
        }
        Set<Integer> intersectionSet = new HashSet<Integer>();
        for (int num : set1) {
            if (set2.contains(num)) {
                intersectionSet.add(num);
            }
        }
        int[] intersection = new int[intersectionSet.size()];
        int index = 0;
        for (int num : intersectionSet) {
            intersection[index++] = num;
        }
        return intersection;
    }
}

思路2：先进行排序，然后使用HashSet，再用二分法查找

public int[] intersection(int[] nums1, int[] nums2) {
        //先对数组进行排序
        Arrays.sort(nums1);
        Set<Integer> resSet = new HashSet<>();
        for (int i = 0; i < nums2.length; i++) {
            //通过二分法查找，判断num2[i]是否存在于num1中
            if (binarySearch(nums1, nums2[i]))
                resSet.add(nums2[i]);
        }

        //这一大坨是把集合set转化为数组
        int[] res = new int[resSet.size()];
        int k = 0;
        for (Integer num : resSet) {
            res[k++] = num;
        }
        return res;
    }

    //二分法查找
    private boolean binarySearch(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            int midValue = nums[mid];
            if (midValue == target) {
                return true;
            } else if (midValue > target) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        return false;
    }