leecode 两个数组的交集 II

import java.util.*;

class test2 {
    public static void main(String[] args) {
        int[] arr1 = { 1, 2, 2, 1 };
        int[] arr2 = { 2, 2 };
        int[] arr3 = intersect1(arr1, arr2);
        System.out.println(Arrays.toString(arr3));

        int[] arr4 = { 1, 1, 2, 2, 3 };
        int[] arr5 = { 1, 3, 5 };
        arr3 = intersect2(arr4, arr5);
        System.out.println(Arrays.toString(arr3));
    }

    // 数组无序排法
    public static int[] intersect1(int[] nums1, int[] nums2) {
        Map<Integer, Integer> counter = new HashMap<>(); // 计数器,key为数组中的数字,value为该数字在数组中出现的次数
        for (int i = 0; i < nums1.length; i++) {
            int num = nums1[i];
            if (counter.containsKey(num)) {
                counter.put(num, counter.get(num) + 1);
            } else {
                counter.put(num, 1);
            }
        }
        List<Integer> tempList = new ArrayList<>();
        for (int i = 0; i < nums2.length; i++) {
            int num = nums2[i];
            if (counter.containsKey(num) && counter.get(num) > 0) {
                counter.put(num, counter.get(num) - 1); // 计数器中记录该数字的次数减1
                tempList.add(num); // 将该数字添加到list中
            }
        }
        int[] result = new int[tempList.size()];
        // 为满足题目返回值类型,将list转换为int数组
        for (int i = 0; i < result.length; i++) {
            result[i] = tempList.get(i);
        }
        return result;
    }

    // 数组有序排法
    public static int[] intersect2(int[] nums1, int[] nums2) {
        int cur1 = 0, cur2 = 0; // 定义指针,指向数组开始位置
        List<Integer> list = new ArrayList<>();
        while (cur1 < nums1.length && cur2 < nums2.length) { // 循环结束条件:任何一个指针指向对应数组的末端
            int num1 = nums1[cur1];
            int num2 = nums2[cur2];
            if (num1 == num2) { // 重复数字,加入结果列表中
                list.add(num1);
                cur1++;
                cur2++;
            } else if (num1 < num2) { // 将cur1指针后移一位,继续下一次比较
                cur1++;
            } else { // 将cur2指针后移一位,继续下一次比较
                cur2++;
            }
        }
        int[] result = new int[list.size()];
        // 为满足题目返回值类型,将list转换为int数组
        for (int i = 0; i < list.size(); i++) {
            result[i] = list.get(i);
        }
        return result;
    }
}

你可能感兴趣的:(leecode)