【LeetCode】704. 二分查找
题意:给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
思考:指明了使用二分法,关键是考虑左右边界的变化。
代码A:
class Solution {
public:
int search(vector<int>& nums, int target) {
// 左闭右闭区间,为了保证循环合法,while应该<=,判断时都要去掉mid
int mid = 0, left = 0, right = nums.size() - 1;
while (left <= right)
{
mid = (left + right) / 2;
if (target < nums[mid])
{
right = mid - 1;
}
else if (target > nums[mid])
{
left = mid + 1;
}
else
{
return mid;
}
}
return -1;
}
};
代码B:
class Solution {
public:
int search(vector<int>& nums, int target) {
// 左闭右开区间,为了保证循环合法,while应该<,判断时注意左边界去掉mid
int mid = 0, left = 0, right = nums.size();
while (left < right)
{
mid = (left + right) / 2;
if (target > nums[mid])
{
left = mid + 1;
}
else if (target < nums[mid])
{
right = mid;
}
else
{
return mid;
}
}
return -1;
}
};
【LeetCode】27. 移除元素
题意:给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路:第一反应是直接遍历调用erase,时间复杂度O(n²)。仔细想想还能用双指针,时间复杂度O(n)。
代码A:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
for (vector<int>::iterator it = nums.begin(); it != nums.end();)
{
if (*it == val)
{
// 避免迭代器失效
it = nums.erase(it);
}
else
{
++it;
}
}
return nums.size();
}
};
代码B:
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];
++slow;
}
}
return slow;
}
};
心态:“第一章 数组part01” 拿下!
参考资料