一、704.二分查找
今天明确了二分查找的两种写法:左闭右闭[left,right]和左闭右开[left,right),两种写法如下:
1.左闭右闭[left,right]
代码如下:
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1;
while (left <= right) {
int mid = left + (right - left)/2;
if (nums[mid] > target)
right = mid - 1;
else if (nums[mid] < target)
left = mid + 1;
else
return mid;
}
return -1;
}
};
其中的关键点为:
1)while (left <= right)
,<=的原因在于,端点是闭区间时,两端点值相同仍然是合法的区间,例如区间[1,1];
2)right = mid - 1
,当nums[mid]大于target时,需要更新左区间的右边界,由于nums[mid]明确大于target,一定不在下一次的搜索区间内,因此右边界更新为mid-1;
3)left = mid + 1
,同理,当nums[mid]小于target时,需要更新右区间的左边界,且nums[mid]不在下一次搜索区间内,因此左边界更新为mid+1;
4)right = nums.size() - 1
,右边为闭区间,右边界为size-1。
2.左闭右开[left,right)
代码如下:
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size();
while (left < right) {
int mid = left + (right - left) / 2;
if (nums[mid] > target)
right = mid;
else if (nums[mid] < target)
left = mid + 1;
else
return mid;
}
return -1;
}
};
其中的关键点为:
1)right = nums.size()
,右边为开区间,右边界为size(因为右端点的值是不取到的);
2)left < right
,区间左闭右开,端点值相同区间不合法,例如[1,1);
3)right = mid
,当nums[mid]大于target时,需要更新左区间的右边界,nums[mid]明确不在下一次搜索范围内,但右边本身是开区间,取不到端点值,故右边界更新为mid;
4)left = mid + 1
,当nums[mid]小于target时,需要更新右区间的左边界,nums[mid]明确不在下一次搜索范围内,因为左边是闭区间,所以左边界更新为mid+1。
在两种方法中,mid = left + (right - left) / 2
,效果等同于mid = (left + right) / 2
,但这种取中间值的写法可以防止溢出。也可以写成mid = left +((right - left)>>1)
(右移一位相当于除以2),运行速度更快。
注意,mid的定义是在while循环内部,因为每次循环都会更新mid的位置,自己写的时候经常会定义在while外面,以后要多留个心眼。
二、27.移除元素
移除元素可以使用暴力求解,也可以使用双指针的方法。今天新学到了双指针的方法,它可以节省一层for循环要做的事。
1.暴力求解
两层for循环,第一层遍历数组寻找需要移除的元素,第二层将需要移除的元素后面的值向前移一位。
代码如下:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int result = nums.size();
for (int i = 0; i < result; i++){
if (nums[i] == val){//发现需要移除的元素时,将后面的数组集体向前移一位
for (int j = i + 1; j < result; j++){
nums[j-1] = nums[j];
}
i--;//因为第二个for循环中把i以后的位置往前移了一位,在移动完后第一个for循环会执行i++,再次开始遍历的位置应该是前一位,所以i也向前移一位
result--;//数组长度减1
}
}
return result;
}
};
2.双指针法
定义:
快指针fastIndex,用来寻找新数组(即删除目标后的数组)里需要的元素;
慢指针slowIndex,用来获取新数组中需要更新的位置(即新数组的下标值)。
两个指针都是在同一个数组上进行操作。
代码如下:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++){
if (nums[fastIndex] != val){//如果快指针指向的不是需要删除的值
nums[slowIndex] = nums[fastIndex];//慢指针也指向这个值,它是新数组需要的元素
slowIndex++;//慢指针移动到下一个位置,等待快指针的判断
}
}
return slowIndex;//结束循环后慢指针的值是新数组的长度,如新数组下标为0-4,慢指针指向的为5
}
};
为什么叫快慢指针呢?
因为只有在nums[fastIndex]不等于val的时候,慢指针才会加(新数组才会更新一个位置);而快指针则是一直往前加,速度是更快的。
双指针法的时间复杂度为O(n),暴力解法为O(n^2),双指针法显然是更优的。
总结
以上是今天对于二分查找和双指针法的学习,二分查找还有【35.搜索插入位置】 和 【34. 在排序数组中查找元素的第一个和最后一个位置】两题拓展,有时间再看一下。