题目描述
给定两个数组 nums1
和 nums2
,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
解题思路
对两个数组进行排序,这样数组nums1可以用二分查找,数组nums2可以直接用判断语句判断是否有相同的元素,在把查到的元素放到动态数组中返回动态数组。
class Solution {
//用nums2中的每个元素去nunms1中进行二分查询
public int[] intersection(int[] nums1, int[] nums2) {
Arrays.sort(nums2);
Arrays.sort(nums1);
ArrayList<Integer> dataArray = new ArrayList<Integer>();
for(int i = 0;i < nums2.length;i++){
if(i == 0 || nums2[i] != nums2[i - 1]){
int a = 0,b = nums1.length;
while( a < b ){
int m = (a + b) >>> 1;
if(nums2[i] < nums1[m]){
b = m;
}else if(nums1[m] < nums2[i]){
a = m + 1;
}else{
dataArray.add(nums2[i]);
break;
}
}
if(a == nums1.length){
break;
}
}
}
int[] result = new int[dataArray.size()];
for (int i = 0; i < result.length; i++) {
result[i] = dataArray.get(i);
}
return result;
}
}
总结
这一题出现的问题有返回值类型问题,由于之前不会动态数组第一次用一开始直接返回dataArray,导致与方法定义的int返回值类型不符。 一开始二分查找没有加break导致运行超时,在之后写程序的时候也会多加考虑。