题目:
给定两个数组,编写一个函数来计算它们的交集。
示例:
代码:
- 解法一
//利用映射实现
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
TreeMap<Integer,Integer> map=new TreeMap<Integer,Integer>();
for(int nums:nums1){ //nums1的键,值放入map中
if(!map.containsKey(nums)){ //不存在时,值为1
map.put(nums,1);
}else{ //存在时 值为当前值+1
map.put(nums,map.get(nums)+1);
}
}
ArrayList<Integer> list=new ArrayList<Integer>();
for(int nums:nums2){
if(map.containsKey(nums)){ //map中有交集元素时
list.add(nums); //将元素加入到list中
map.put(nums,map.get(nums)-1); //将其值-1
if(map.get(nums)==0){ //值为0 则map中没有这个键 删除键
map.remove(nums);
}
}
}
int[] res=new int[list.size()];
for(int i=0;i<list.size();i++){
res[i]=list.get(i);
}
return res;
}
}
- 别人的代码
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
int i=0,j=0;
Arrays.sort(nums1);
Arrays.sort(nums2);
int minle=nums1.length<nums2.length?nums1.length:nums2.length;
int[] res=new int[minle];
int k=0;
while( i < nums1. length && j < nums2. length) {
if(nums1[i] == nums2[j]){ //如果相等 将元素加入res中
res[k++] = nums1[i++];
j++; //j++ 更新步长 为什么不i++ j++因为有可能出现元素重复出现的情况
}else{
if(nums1[i] > nums2[j]){ //因为数组是排序过的 j++
j++;
}
else{ //排过序 i--
i++;
}
}
}
return Arrays.copyOfRange(res,0,k);
}
}