702 二分查找
1、做出假设:
提示:
- 你可以假设 nums 中的所有元素是不重复的。
- n 将在 [1, 10000]之间。
- nums 的每个元素都将在 [-9999, 9999]之间。
2、输入输出:
示例 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
3、解决方案
二分查找涉及的很多的边界条件,逻辑比较简单,但就是写不好。例如到底是 while(left < right)
还是 while(left <= right)
,到底是right = middle
呢,还是要right = middle - 1
呢?
大家写二分法经常写乱,主要是因为对区间的定义没有想清楚,区间的定义就是不变量。要在二分查找的过程中,保持不变量,就是在while寻找中每一次边界的处理都要坚持根据区间的定义来操作,这就是循环不变量规则。
3.1、数组下标区间左闭右闭
#include <iostream>
#include <vector>
using namespace std;
// 版本一
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效,所以用 <=
int middle = left + ((right - left) / 2); // 防止溢出 等同于(left + right) / 2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
}
else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
}
else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
};
int main() {
vector<int> nums = { -1,0,3,5,9,12 }; // 给定的有序数组
int target = 9; // 目标值
Solution solution; //实例化
int result = solution.search(nums, target); // 调用solution类中的二分查找函数
if (result != -1) {
cout << "Target found at index: " << result << endl;
}
else {
cout << "Target not found in the array." << endl;
}
return 0;
}
3.2、数组下标区间左闭右开
#include <iostream>
#include <vector>
using namespace std;
// 版本二
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size(); // 定义target在左闭右开的区间里,即:[left, right)
while (left < right) { // 因为left == right的时候,在[left, right)是无效的空间,所以使用 <
int middle = left + ((right - left) >> 1); // 使用右移操作 >> 1 取代除法 / 2 是一种优化方法
if (nums[middle] > target) {
right = middle; // target 在左区间,在[left, middle)中
}
else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,在[middle + 1, right)中
}
else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
};
int main() {
vector<int> nums = { -1,0,3,5,9,12 }; // 给定的有序数组
int target = 9; // 目标值
Solution solution; //实例化
int result = solution.search(nums, target); // 调用solution类中的二分查找函数
if (result != -1) {
cout << "Target found at index: " << result << endl;
}
else {
cout << "Target not found in the array." << endl;
}
return 0;
}
27. 移除元素
题目假设
示例 1: 输入:nums = [3,2,2,3], val = 3 输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。
例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例 2: 输入:nums = [0,1,2,2,3,0,4,2], val = 2 输出:5, nums = [0,1,3,0,4]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
1、暴力解法
#include <iostream>
#include <vector>
using namespace std;
// 时间复杂度:O(n^2)
// 空间复杂度:O(1)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++) {
if (nums[i] == val) { // 发现需要移除的元素,就将数组集体向前移动一位,覆盖移除元素的位置
for (int j = i + 1; j < size; j++) {
nums[j - 1] = nums[j];
}
i--; // 因为下标i以后的数值都向前移动了一位,所以i也向前移动一位
size--; // 此时数组的大小-1
}
}
return size;
}
};
// 函数用于输出 vector 中的元素
void printVector(const std::vector<int>& vec) {
std::cout << "nums = [";
for (size_t i = 0; i < vec.size(); ++i) {
std::cout << vec[i];
if (i != vec.size() - 1) { // 添加 ,号
std::cout << ", ";
}
}
std::cout << "]" << std::endl;
}
int main() {
vector<int> nums = { 0,1,2,2,3,0,4,2 }; // 给定的有序数组
int val = 2; // 目标值
Solution solution; // 实例化
int result = solution.removeElement(nums, val); // 调用solution类中的二分查找函数
// 调用函数输出 vector 中的元素
printVector(nums);
// 输出nums中的元素个数
cout << "size: " << result << endl;
return 0;
}
2、双指针(快慢指针)
方式一:
// 时间复杂度:O(n)
// 空间复杂度:O(1)
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slowIndex = 0;
for (int fastIndex = 0; fastIndex < nums.size(); fastIndex++) {
if (val != nums[fastIndex]) {
nums[slowIndex++] = nums[fastIndex];
}
}
return slowIndex;
}
};
方式二:
/**
* 相向双指针方法,基于元素顺序可以改变的题目描述改变了元素相对位置,确保了移动最少元素
* 时间复杂度:O(n)
* 空间复杂度:O(1)
*/
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int leftIndex = 0;
int rightIndex = nums.size() - 1;
while (leftIndex <= rightIndex) {
// 找左边等于val的元素
while (leftIndex <= rightIndex && nums[leftIndex] != val) {
++leftIndex;
}
// 找右边不等于val的元素
while (leftIndex <= rightIndex && nums[rightIndex] == val) {
--rightIndex;
}
// 将右边不等于val的元素覆盖左边等于val的元素
if (leftIndex < rightIndex) {
nums[leftIndex++] = nums[rightIndex--];
}
}
return leftIndex; // leftIndex一定指向了最终数组末尾的下一个元素
}
};