CV炼丹师勇闯力扣训练营
代码随想录算法训练营第一天| 704. 二分查找、27. 移除元素
一、704 二分查找
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
【注意】使用二分查找的前提:有序数组、且无重复元素
【难点】边界问题,要保证左闭右闭区、左闭右开两种区间定义始终有意义,即确保不变量始终成立
代码如下(C++):
// 左闭右闭
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;
}
};
// 左闭右开
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);
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;
}
};
代码如下(Python):
# 左闭右闭
class Solution:
def search(self, nums, tar) -> int:
left, right = 0, len(nums) - 1
while(left <= right):
mid = left + ((right - left) >> 1)
if(nums[mid] > tar):
right = mid - 1
elif(nums[mid] < tar):
left = mid + 1
else:
return mid
return -1
# 左闭右开
class Solution:
def search(self, nums, tar):
left, right = 0, len(nums)
while(left <= right):
mid = left + ((right - left) >> 1)
if(nums[mid] > tar):
right = mid # 右开不取值,所以下一次循环right为mid
elif(nums[mid] < tar):
left = mid + 1 # 左闭取值,所以下一次循环left不能算上mid,得从mid+1开始
else:
return mid
return -1
二、27 移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:
- 更改 nums 数组,使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
- 返回 k
暴力解法:两层for循环,第⼀个for循环遍历数组元素 ,第⼆个for循环更新数组
☆ 双指针法(快慢指针法): 通过⼀个快指针和慢指针在⼀个for循环下完成两个for循环的⼯作。
快指针:寻找新数组的元素 ,新数组就是不含有⽬标元素的数组
慢指针:指向更新 新数组下标的位置,最终指向新数组尾地址,就是新数组的大小
![双指针操作](https://img-blog.csdnimg.cn/direct/9f7b2ecfe7834769beffd57cee8847b9.png)
引用代码随想录中的一张动图,能够直观看出快慢指针的运行流程
代码如下(C++):
// 时间复杂度: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;
}
};
代码如下(Python):
class Solution:
def removeElement(self, nums, val) -> int:
# 快慢指针
fast = 0 # 快指针
slow = 0 # 慢指针
size = len(nums)
while fast < size: # 不加等于是因为,a = size 时,nums[a] 会越界
# slow 用来收集不等于 val 的值,如果 fast 对应值不等于 val,则把它与 slow 替换
if nums[fast] != val:
nums[slow] = nums[fast]
slow += 1 # 【注意】Python中没有++
fast += 1
return slow
总结
- 二分查找的两种区间
- 用快慢指针来处理数组元素移除问题