LeetCode之Intersection of Two Arrays

1、题目

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

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

Note:

• Each element in the result must be unique.
• The result can be in any order.

Subscribe to see which companies asked this question.

2、代码实现

public class Solution {
      public int[] intersection(int[] nums1, int[] nums2) {
     if (nums1 == null || nums2 == null) {
     return null;
     }
     Map<Integer, Integer> map = new HashMap<Integer, Integer>();
     List<Integer> list = new ArrayList<Integer>();
     for (int  i = 0; i < nums1.length; ++i) {
     map.put(nums1[i], 1);
     }
     for (int i = 0; i < nums2.length; ++i) {
     if (map.get(nums2[i]) != null && map.get(nums2[i]) != 2) {
     list.add(nums2[i]);
     map.put(nums2[i], 2);
     }
     }
     int a[] = new int[list.size()];
     for(int i = 0, j = list.size(); i < j; i++){
     a[i] = list.get(i);
     }
 return a;
 }
}

 

 

3、总结

一开始写的时候是这样写的

if (map.get(nums2[i]) != null) {
 list.add(nums2[i]);
 }

这样写的话，在第二个数组里面出现了多个交集元素，就会得到结果很多一样的交集元素，很明显，错了，

以后写的时候需要注意，保证数据的唯一性，做个开关，得到了就关闭，入股哦后面有相同元素出现的时候，门已经关了，就添加不进去了，保证数据的唯一性。