代码随想录算法训练营第一天 | 704.二分查找、27.移除元素*
704.二分查找
题目:给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
class Solution {
public:
int search(vector<int>& nums, int target) {
// 方法一,左闭右闭
int left = 0;
int 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;
}
};
class Solution {
public:
int search(vector<int>& nums, int target) {
// 方法二,左闭右开
int left = 0;
int 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;
}
};
二分查找的前提是:有序数组。时间复杂度O(logn)。
利用区间不变量来考虑边界范围,区间不变量意思就是说每次分区间时,得到的新区间继续采用左闭右闭/左闭右开进行划分。
27.移除元素
题目:给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
注意:数组的元素在内存地址中是连续的,不能单独删除数组中的某个元素,只能覆盖。
// 暴力解法 O(n^2)
// 第一层for循环用于查找到需要删除的元素
// 第二层用于更新数组。
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--;
size--;
}
}
return size;
}
};
// 快慢指针 O(n)
// 快指针定义:遍历寻找新数组的元素,即寻找不是待删除元素的其他元素
// 用其他元素覆盖待删除元素
// 慢指针定义: 指向的是覆盖后新数组的下标,,最后返回solwIndex就是数组的长度
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;
}
};