分析:可以采用set数据结构,来遍历数组1,将数组1中的元素放入set中,再遍历数组2,如果set中存在的话,就放入另外一个set中,最后将另外一个set中的元素放入数组并返回即可。
public class Solution {
/*
* @param nums1: an integer array
* @param nums2: an integer array
* @return: an integer array
*/
public int[] intersection(int[] nums1, int[] nums2) {
// write your code here
if(nums1.length==0){
return nums1;
}else if(nums2.length==0){
return nums2;
}
Set<Integer> set=new HashSet<>();
for(Integer s:nums1){
set.add(s);
}
Set<Integer> res=new HashSet<>();
for(int i=0;i<nums2.length;i++){
if(set.contains(nums2[i])){
res.add(nums2[i]);
}
}
int[] result=new int[res.size()];
int index=0;
for(Integer s:res){
result[index++]=s;
}
return result;
}
};
本题与上一题的区别在于返回的交集包不包含重复元素。如果包含重复元素的话,遍历数组1中所有元素的时候应该采用hashmap数据结构,key为数组元素,value为该数组元素出现的次数。遍历数组2中元素时,如果hashmap中存在该元素且该key的value值大于0,则value--,并将该元素放入数组中,否则continue。最后返回数组。
public class Solution {
/*
* @param nums1: an integer array
* @param nums2: an integer array
* @return: an integer array
*/
public int[] intersection(int[] nums1, int[] nums2) {
// write your code here
if(nums1.length==0){
return nums1;
}else if(nums2.length==0){
return nums2;
}
HashMap<Integer,Integer> hashMap=new HashMap<>();
for(Integer s:nums1){
if(hashMap.containsKey(s)){
int value=hashMap.get(s);
hashMap.put(s,++value);
}else{
hashMap.put(s,1);
}
}
ArrayList<Integer> arr=new ArrayList<>();
for(Integer s:nums2){
if(hashMap.containsKey(s) && hashMap.get(s)>0){
int value=hashMap.get(s);
hashMap.put(s,--value);
arr.add(s);
}else{
continue;
}
}
int[] res=new int[arr.size()];
int index=0;
for(Integer s:arr){
res[index++]=s;
}
return res;
}
};