1. 程式人生 > >找出兩個數組中不相同的元素

找出兩個數組中不相同的元素

不同的 out 優化 ont 相同 array ted str args

找出兩個數組中不相同的元素,網上貼出的代碼基本如下:

/**
 * 找出兩個數組中不同的元素
 */
public class Test3 {
    public static Set<Integer> compare(Integer[] a, Integer[] b) {
        Set<Integer> aSet = new HashSet<>();
        Set<Integer> diffSet = new HashSet<>();
        aSet.addAll(Arrays.asList(a));
        for (Integer i : b) {
            if(!aSet.contains(i)) {
                diffSet.add(i);
            }
        }
        return diffSet;
    }

    public static void main(String[] args) throws InterruptedException {
        Integer[] a = {1, 2, 3};
        Integer[] b = {1, 2, 3, 4};
        System.out.println(compare(a, b));
    }
}

輸出

[4]

  

猜猜有什麽坑??

Integer[] a = {0, 1, 2, 3};
Integer[] b = {1, 2, 3, 4};

輸出
[4]  

改進後的代碼

/**
 * 找出兩個數組中不同的元素
 */
public class Test3 {
    public static Set<Integer> compare(Integer[] a, Integer[] b) {
        Set<Integer> aSet = new HashSet<>();
        Set<Integer> diffSet = new HashSet<>();
        aSet.addAll(Arrays.asList(a));
        for (Integer i : b) {
            if(!aSet.contains(i)) {
                diffSet.add(i);
            }
        }
        return diffSet;
    }

    public static void main(String[] args) throws InterruptedException {
        Integer[] a = {0, 1, 2, 3};
        Integer[] b = {1, 2, 3, 4};
        Set<Integer> diffSetA = compare(a, b);
        Set<Integer> diffSetB = compare(b, a);
        diffSetA.addAll(diffSetB);
        System.out.println(diffSetA);
    }
}

  輸出

[0, 4]

  要調用兩次方法,有點啰嗦,再次優化如下:

/**
 * 找出兩個數組中不同的元素
 */
public class Test4 {
    public static Set<Integer> compare(Integer[] a, Integer[] b) {
        // 存儲所有元素
        Set<Integer> allSet = new HashSet<>();
        // 存儲相同的元素
        Set<Integer> sameSet = new HashSet<>();
        allSet.addAll(Arrays.asList(a));
        for (Integer i : b) {
            if(allSet.contains(i)) {
                sameSet.add(i);
            }
        }
        allSet.addAll(Arrays.asList(b));
        // 移除相同的元素
        allSet.removeAll(sameSet);
        return allSet;
    }

    public static void main(String[] args) throws InterruptedException {
        Integer[] a = {0, 1, 2, 3};
        Integer[] b = {1, 2, 3, 4};
        System.out.println(compare(a, b));
    }
}

  打印

[0, 4]

  

找出兩個數組中不相同的元素