G面经prepare: set difference

 

给你A{1,2,3,4,4,5}， B{2,4}，求A-B={1,3,4,5},很简单. visit 1

只用一个HashMap

package TwoSets;
import java.util.*;

public class Solution {
    public ArrayList<Integer> findDiff(int[] arr1, int[] arr2) {
        ArrayList<Integer> res = new ArrayList<Integer>();
        HashMap<Integer, Integer> m1 = new HashMap<Integer, Integer>();
        for (int item1 : arr1) {
            if (!m1.containsKey(item1)) {
                m1.put(item1, 1);
            }
            else {
                m1.put(item1, m1.get(item1)+1);
            }
        }
        for (int item2 : arr2) {
            if (m1.containsKey(item2)) {
                m1.put(item2, m1.get(item2)-1);
            }
            if (m1.get(item2) == 0) m1.remove(item2);
        }
        for (int elem : m1.keySet()) {
            int num = m1.get(elem);
            while (num > 0) {
                res.add(elem);
                num--;
            }
        }
        return res;
    }
    

    /**
     * @param args
     */
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Solution sol = new Solution();
        ArrayList<Integer> res = sol.findDiff(new int[]{1,2,3,4,4,5}, new int[]{2,4});
        System.out.println(res);
    }

}