二分查找
题目描述
给定一个 n
个元素有序的(升序)整型数组 nums
和一个目标值 target
,写一个函数搜索 nums
中的 target
,如果目标值存在返回下标,否则返回 -1
。
示例1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
示例2:
输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
解题思路
题目中提到了有序数组,且数组中无重复元素,因此可以采用二分查找的方式。
二分查找共有两种写法,分别为左闭右闭和左闭右开。
写法一:左闭右闭([left,right])
-
初始值:left = 0,right = nums.size() - 1
-
查找条件:while(left <= right),使得[left,right]有意义
-
二分流程
- nums[mid] > target时,说明target在左区间,由于target一定不等于nums,因此左区间下标结束位置应该设置为mid - 1
- nums[mid] < target时,说明target在右区间,由于target一定不等于nums,因此右区间下标起始位置应该设置为mid + 1
写法二:左闭右开([left,right))
-
初始值:left = 0,right = nums.size()
-
查找条件:while(left < right),使得[left,right)有意义
-
二分流程
- nums[mid] > target时,说明target在左区间,寻找区间是左闭右开区间,所以right更新为middle
- nums[mid] < target时,说明target在右区间,由于target一定不等于nums,因此右区间下标起始位置应该设置为mid + 1
时空复杂度
- 时间复杂度:O(log n)
- 空间复杂度:O(1)
代码实现
测试地址:https://leetcode.cn/problems/binary-search/
写法一:左闭右闭
class Solution {
public:
int search(vector<int>& nums, int target) {
// 左闭右闭
int left = 0, right = nums.size() - 1;
while (left <= right) {
int mid = (left + right) / 2;
if (target < nums[mid]) {
right = mid - 1;
} else if (target > nums[mid]) {
left = mid + 1;
} else {
return mid;
}
}
return -1;
}
};
写法二:左闭右开
class Solution {
public:
int search(vector<int>& nums, int target) {
// 左闭右开
int left = 0, right = nums.size();
while (left < right) {
int mid = (left + right) / 2;
if (target < nums[mid]) {
right = mid;
} else if (target > nums[mid]) {
left = mid + 1;
} else {
return mid;
}
}
return -1;
}
};
移出元素
题目描述
给你一个数组 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。
解题思路
由于数组的元素在内存中是连续的,因此在删除某个元素后,只能进行数组元素覆盖的方式实现这个目的。
暴力解法
总体思路:
使用两层for循环,第一层遍历数组元素,找到元素对应下标,第二层循环将当前元素后面的元素进行依次覆盖,实现移出元素的目的。
时空复杂度分析:
-
时间复杂度:O(n^2)
-
空间复杂度:O(1)
双指针
总体思路:
使用快慢指针在一层for循环下,完成暴力解法两层for循环的工作。
其中快指针指向的元素为新数组中包含的元素,即不等于val的元素
慢指针指向的是新数组的下标位置
通过快指针遍历数组元素,只要快指针指向的元素值不等于val,那么同时移动慢指针下标赋值操作,如果快指针的下标恰好指向了val对应的元素,那么跳过慢指针的赋值操作,直至快指针下标继续移动到不为val元素为止。
时空复杂度分析
-
时间复杂度:O(n)
-
空间复杂度:O(1)
代码实现
测试地址:https://leetcode.cn/problems/remove-element/
暴力解法:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++) {
if (val == nums[i]) { // 发现需要移除的元素,就将数组集体向前移动一位
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
i--; // 因为下标i以后的数值都向前移动了一位,所以i也向前移动一位
size--;
}
}
return size;
}
};
双指针:
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;
}
};