704.二分查找法
一、题目要求
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
二、解题思路
学过C语言都知道,在一个定义好的数组中寻找一个数,直接使用一个for循环遍历一遍就可以找到目标数据的下标,但是这样的空间和时间复杂度都是O(n)。既然是算法,就应该有算法的样子!
这道题目的前提是数组为有序数组,同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下标可能不是唯一的,这些都是使用二分法的前提条件,当大家看到题目描述满足如上条件的时候,可要想一想是不是可以用二分法了。
二分法也就是数据结构学的折半查找法,既然是折半,所以是每一次取一半来找,所以时间复杂度就是O(log n)。那怎么样定义“取一半”呢?这个时候就要用到游标,定义一个tag,比如middle代表中间元素的下标,初始定义left为数组的0位置,定义right为数组的长度减一(下标从0开始),那么准备工作就准备好了!
三、两种写法
3.1 左闭右闭
第一种写法,我们定义 target 是在一个在左闭右闭的区间里,也就是[left, right] (这个很重要非常重要)。区间的定义这就决定了二分法的代码应该如何写,因为定义target在[left, right]区间,所以有如下两点:
- while (left <= right) 要使用 <= ,因为left == right是有意义的,所以使用 <=(这里是因为是右闭区间所以取得到!)
- if (nums[middle] > target) ,在这种情况下,target在左半边的区间,所以right 要赋值为 middle - 1,因为当前这个nums[middle]一定不是target,那么接下来要查找的左区间结束下标位置就是 middle - 1
代码实现:
class Solution {
public int search(int[] nums, int target) {
// 避免当 target 小于nums[0] nums[nums.length - 1]时多次循环运算
if (target < nums[0] || target > nums[nums.length - 1]) {
return -1;
}
int left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + ((right - left) >> 1);
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
left = mid + 1; // target 在右区间,在[middle + 1, right)中
else if (nums[mid] > target)
right = mid - 1;
}
return -1;
}
}
3.2左闭右开
如果说定义 target 是在一个在左闭右开的区间里,也就是[left, right) ,那么二分法的边界处理方式则截然不同。
有如下两点:
- while (left < right),这里使用 < ,因为left == right在区间[left, right)是没有意义的
- if (nums[middle] > target) right 更新为 middle,因为当前nums[middle]不等于target,去左区间继续寻找,而寻找区间是左闭右开区间,所以right更新为middle,即:下一个查询区间不会去比较nums[middle]
代码实现:
class Solution {
public int search(int[] nums, int target) {
int left = 0, right = nums.length;
while (left < right) { // 因为left == right的时候,在[left, right)是无效的空间,所以使用 <
int mid = left + ((right - left) >> 1);
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
left = mid + 1; // target 在右区间,在[middle + 1, right)中
else if (nums[mid] > target)
right = mid; // target 在左区间,在[left, middle)中 ,right直接取就可以(右开区间)!
}
return -1;
}
}
27. 移除元素
一、题目要求
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。
示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
你不需要考虑数组中超出新长度后面的元素。
二、解题思路
第一想法就是,循环遍历,比较,移位。就是暴力的解法,但是这样复杂度是O()。
第二个方法是双指针,当时没有想到,主要是两个index改变数组的元素位置就可以实现。
三、代码实现
3.1 暴力解法
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int num_size = nums.size();
for(int i = 0;i < num_size; i++){
if(nums[i] ==val){
for(int j = i+1;j< num_size;j++){ //这里j=i+1写错了, 取i位置后面的一位
nums[j-1] = nums[j]; //这里循环里面只是对j做操作!
}
num_size--;
i--;
}
}
return num_size;
}
};
3.2 双指针法
正向思路:(后面是有fast自增的操作,相当于找到了就跳过!)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slow = 0;
for(int fast =0; fast < nums.size(); fast++ ){
if(nums[fast] == val){
}else{
nums[slow++] = nums[fast];
}
}
return slow;
}
};
修改一下:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slow = 0;
for(int fast =0; fast < nums.size(); fast++ ){
if(nums[fast] != val){
nums[slow++] = nums[fast];
}
}
return slow;
}
};