LeetCode-350-兩個陣列的交集 II

兩個陣列的交集 II

題目描述：給定兩個陣列，編寫一個函式來計算它們的交集。

示例說明請見LeetCode官網。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：陣列遍歷

• 首先，宣告一個Map為firstMap，其中key為出現的數字，value為相應數字出現的次數，然後遍歷nums1，將數字和出現的次數初始化到firstMap中；
• 然後，宣告一個陣列為result；
• 然後遍歷nums2中的數字，判斷如果firstMap的key中包含該數字並且對應的次數大於0，則將這個數字放到result中。
• 遍歷完成後，返回result所有的數字即為結果。

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class LeetCode_350 {
    public static int[] intersect(int[] nums1, int[] nums2) {
        // 第一個陣列中的數字以及出現的次數
        Map<Integer, Integer> firstMap = new HashMap<>();
        for (int i : nums1) {
            if (firstMap.containsKey(i)) {
                firstMap.put(i, firstMap.get(i) + 1);
            } else {
                firstMap.put(i, 1);
            }
        }

        int[] result = new int[nums1.length];
        int index = 0;
        for (int i : nums2) {
            if (firstMap.containsKey(i) && firstMap.get(i) > 0) {
                result[index++] = i;
                firstMap.put(i, firstMap.get(i) - 1);
            }
        }

        return Arrays.copyOfRange(result, 0, index);
    }

    public static void main(String[] args) {
        int[] nums1 = new int[]{1, 2, 2, 1}, nums2 = new int[]{2, 2};
        for (int i : intersect(nums1, nums2)) {
            System.out.print(i + " ");
        }
    }
}

【每日寄語】 用你的笑容去改變這個世界，別讓這個世界改變了你的笑容。