LeetCode-349-兩個陣列的交集

兩個陣列的交集

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

示例說明請見LeetCode官網。

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

解法一：陣列遍歷

• 首先將nums1中的所有數字放入Set中firstNums；
• 然後初始化一個result陣列放最終的結果，並宣告一個Set為secondNums用來放nums2和nums1重複的元素，然後，遍歷nums2；
• 如果firstNums中存在且secondNums不存在，就將當前的數放入result中，並且放入secondNums中。
• 遍歷完成後，返回result中的所有數字。

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

public class LeetCode_349 {
    public static int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> firstNums = new HashSet<>();
        for (int i : nums1) {
            firstNums.add(i);
        }
        int[] result = new int[nums1.length];
        int index = 0;
        Set<Integer> secondNums = new HashSet<>();
        for (int i : nums2) {
            if (firstNums.contains(i) && !secondNums.contains(i)) {
                secondNums.add(i);
                result[index++] = i;
            }
        }

        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 : intersection(nums1, nums2)) {
            System.out.print(i + " ");
        }
    }
}

【每日寄語】 世上最耀眼的光芒除了太陽還有你努力的模樣。