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]);
}
这样写的话,在第二个数组里面出现了多个交集元素,就会得到结果很多一样的交集元素,很明显,错了,
以后写的时候需要注意,保证数据的唯一性,做个开关,得到了就关闭,入股哦后面有相同元素出现的时候,门已经关了,就添加不进去了,保证数据的唯一性。