349, 350. Intersection of Two Arrays | & ||

Intersection of Two Arrays

Given two arrays, write a function to compute their intersection.

Example

Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

分析：

这种考察一个数是否存在另一个集合中，一般来讲，都会用到HashSet. 如果不允许用额外的存储空间，可以对nums1排序，然后使用二分查找。

public class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        if (nums1 == null || nums2 == null || nums1.length == 0 || nums2.length == 0) {
            int[] arr = new int[0];
            return arr;
        }
        
        Set<Integer> set = new HashSet<>();
        for (int i : nums1) {
            set.add(i);
        }
        
        List<Integer>  list = new ArrayList<>();
        
        for (int i : nums2) {
            if (set.contains(i)) {
                list.add(i);
                set.remove(i);
            }
        }
        
        return list.stream().mapToInt(i->i).toArray();
    }

Intersection of Two Arrays II

Given two arrays, write a function to compute their intersection.

Example

Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2, 2].

这一次需要把所有的都找出来，可以用HashMap<Integer, Integer> 来存储，第一个Integer指的是数，第二个指的是那个数存在的个数。

public class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        if (nums1 == null || nums2 == null || nums1.length == 0 || nums2.length == 0) {
            int[] arr = new int[0];
            return arr;
        }
        
        Map<Integer, Integer> map = new HashMap<>();
        for (int i : nums1) {
            map.put(i, map.getOrDefault(i, 0) + 1);
        }
        
        ArrayList<Integer>  list = new ArrayList<>();
        
        for (int i : nums2) {
            if (map.containsKey(i) && map.get(i) >= 1) {
                list.add(i);
                map.put(i, map.get(i) - 1);
            }
        }
        return list.stream().mapToInt(i->i).toArray();
    }
}

 

转载请注明出处：cnblogs.com/beiyeqingteng/