排序和搜索
在本节的练习题中,只有两道题目,涉及到的算法有:数组合并、二分查找
1.合并两个有序数组
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
- 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
- 你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
langage Java
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
}
}
思路
要求是将数组合并到nums1中,首先判断nums1是否为空,如果为空,那么将nums2复制到nums1中即可。
- i指针指向nums1,初始化时指向nums1的第一个元素
- j指针指向nums2,初始化时指向nums2的第一个元素
- 比较i、j指针指向的值的大小,将较小值存入新数组,并将其指针指向后移,该操作一直进行,直至两个指针有一个越界
- 将未比较完全的数组的剩余部分复制到新数组中
- 将新数组内容复制到nums1中
解法
java 解法
class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {
if(n==0) return;
//nums1为空,复制nums2至nums1
if(m==0){
for(int i=0;i<n;i++){
nums1[i]=nums2[i];
}
return;
}
int[] a=new int[m+n];
int i=0,j=0;
int t=0;
//比较,填充新数组
while(i<m&&j<n){
if(nums1[i]<nums2[j]) a[t++]=nums1[i++];
else a[t++]=nums2[j++];
}
//复制剩余部分
if(i>=m){
for(;j<n;j++){
a[t++]=nums2[j];
}
}
if(j>=n){
for(;i<m;i++){
a[t++]=nums1[i];
}
}
//复制新数组内容至nums1
for(int q=0;q<n+m;q++){
nums1[q]=a[q];
}
}
}
2.第一个错误的版本
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
示例:
给定 n = 5,并且 version = 4 是第一个错误的版本。
调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true
所以,4 是第一个错误的版本。
langage Java
/* The isBadVersion API is defined in the parent class VersionControl.
boolean isBadVersion(int version); */
public class Solution extends VersionControl {
public int firstBadVersion(int n) {
}
}
思路
基本的思想是基于二分查找
- 如果中间版本是错误版本,那么high=mid-1,继续二分查找
- 如果中间版本是正确版本,那么low=mid+1,继续二分查找
- 当low>high时,停止循环
这里有一个非常重要的点,先看我第一次的做法:
public int firstBadVersion(int n) {
if (n <= 0) return 0;
int low = 1;
int high = n;
while (low <= high) {
int mid = (high + low) / 2;
if (isBadVersion(mid)) high = mid - 1;
else low = mid + 1;
}
return low;
}
逻辑上是没有错误的,但是在提交的时候遇到了错误,引起错误的测试集为:
2126753390
1702766719
解释一下:第一行是总版本号n,第二行是第一个错误的版本号
经调试打印出的数据如下:
System.out.println(low+","+high+"," + mid);
97627,2126753390,-736558139
-736558138,2126753390,695097626
695097627,2126753390,-736558139
-736558138,2126753390,695097626
695097627,2126753390,-736558139
-736558138,2126753390,695097626
695097627,2126753390,-736558139
-736558138,2126753390,695097626
可以看到mid值超出了int能表示的大小,所以出现了负数
本来是这样计算mid的,但是两数相加可能会超出Int的范围
int mid = (high + low) / 2;
所以使用以下方法计算mid值:
int mid = low + (high - low) / 2;
解法
java 解法
/* The isBadVersion API is defined in the parent class VersionControl.
boolean isBadVersion(int version); */
public class Solution extends VersionControl {
public int firstBadVersion(int n) {
if(n <= 0) {
return 0;
}
int low = 1, high = n;
while(low <= high) {
int mid = low + (high - low) / 2;
if(isBadVersion(mid)) {
high = mid - 1;
}
else {
low = mid + 1;
}
}
return low;
}
}