代码随想录算法训练营第一天| 704. 二分查找、27. 移除元素
题目一
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
题目链接
自己的解法
class Solution
{
public:
int search(vector<int> &nums, int target)
{
for (int i = 0; i < nums.size(); i++)
{
if (nums[i] == target)
return i;
}
return -1;
}
};
仔细想想,问题还是很多的,首先:时间复杂度为O(n^2),其次是数组是有序的,且无重复元素这个条件根本没有用上。
二分法解法
故此,详细查看二分法的解法:
首先,二分法有两个情况:左闭右闭区间[left,right]和左闭右开区间[left,right)
第一种情况(左闭右闭):
while(left<=right)//闭区间特性
if(nums[middle] > target)
right = middle - 1;//middle一定不是target
详细代码:
class Solution
{
public:
int search(vector<int> &nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left <= right)
{
int middle = (left + right) / 2;
if (nums[middle] > target)
right = middle - 1;
else if (nums[middle] < target)
left = middle + 1;
else
return middle;
}
return -1;
}
};
第二种情况(左闭右开):
while(left < right) //left = right没有意义
if(num[middle] > target)
right = middle;
详细代码:
class Solution
{
public:
int search(vector<int> &nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left < right)
{
int middle = (left + right) / 2;
if (nums[middle] > target)
right = middle;
else if (nums[middle] < target)
left = middle + 1;
else
return middle;
}
return -1;
}
};
这样的时间复杂度为O(logn)
思考
边界的意义是否必要遵守?
如果我想图省事,比如好记忆,省去了+1/-1的操作是否会导致数值错误
代码如下:
class Solution
{
public:
int search(vector<int> &nums, int target)
{
int left = 0;
int right = nums.size() - 1;
while (left < right)
{
int middle = (left + right) / 2;
if (nums[middle] > target)
right = middle;
else if (nums[middle] < target)
left = middle;
else
return middle;
}
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。
你不需要考虑数组中超出新长度后面的元素。
题目链接
自己的解法(暴力解法)
思考:
由于限制了空间复杂度为O(1),因此无法使用新的容器,并仅仅只需要返回长度,由此有以下解法:
class Solution
{
public:
int removeElement(vector<int> &nums, int val)
{
int size = nums.size();
for (int i = 0; i < nums.size(); i++)
{
if (nums[i] == val)
{
size -= 1;
for (int j = i; j < nums.size() - 1; j++)
{
nums[j] = nums[j++];
}
}
}
return size;
}
};
输入:1, 2, 7, 7, 7, 9, 9
输出:1,2,7,7,7,7,9
问题出在:
首先:i的长度减少了,因此i需要减1
其次:j应当小于size而不是nums.size()
修正后的代码:
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;
}
};
这种解法居然容易超时
故看看随想录解法:
双指针法
通过两个指针(一个快一个慢)的方式完成两个循环
快指针负责调节数组,慢指针用于返回size
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;
}
};
这个写的太巧妙了,可以说把能省的步骤都省了,看的我意犹未尽