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]);
	 }

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

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