2215. Find the Difference of Two Arrays - Easy

前往題目

想法

• 用兩個set，各自裝nums1和nums2的元素，然後利用.contains來查看另一個是否也有，沒有的話就加到答案裡

思路

同想法

Code

class Solution {
    public List<List<Integer>> findDifference(int[] nums1, int[] nums2) {
        
        List<List<Integer>> res = new ArrayList<>();
        res.add(new ArrayList<>());
        res.add(new ArrayList<>());
        Set<Integer> nums1Set = new HashSet();
        Set<Integer> nums2Set = new HashSet();

        // 加入set
        for (int num : nums1) {
            nums1Set.add(num);
        }

        for (int num : nums2) {
            nums2Set.add(num);
        }

        // 判斷哪些數字是nums1獨有的
        for (int num : nums1Set) {
            if (!nums2Set.contains(num)) res.get(0).add(num);
        }

        for (int num : nums2Set) {
            if (!nums1Set.contains(num)) res.get(1).add(num);
        }

        return res;
    }
}