704. 二分查找
给定一个 n
个元素有序的(升序)整型数组 nums
和一个目标值 target
,写一个函数搜索 nums
中的 target
,如果目标值存在返回下标,否则返回 -1
。
题目链接:力扣
思路与解答:
这道题的关键在于分清楚边界值,左闭右闭和左闭右开的每次操作不同,
对于左闭右闭,left = mid + 1; right = mid - 1;
时间复杂度:O(log n)
空间复杂度:O(1)
代码如下:
//左闭右闭
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1, mid = (left + right) / 2;
while(left <= right)
{
if(target > nums[mid])
left = mid + 1;
else if(target < nums[mid])
right = mid - 1;
else
return mid;
mid = (left + right) / 2;
}
return -1;
}
};
对于左闭右开,left = mid + 1; right = mid;
时间复杂度:O(log n)
空间复杂度:O(1)
代码如下:
//左闭右开
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size(), mid = (left + right) / 2;
while(left < right)
{
if(target > nums[mid])
left = mid + 1;
else if(target < nums[mid])
right = mid;
else
return mid;
mid = (left + right) / 2;
}
return -1;
}
};
27.移除数组元素
给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
题目链接:力扣
思路与解答:
这道题可以用暴力求解,要删除哪个元素就把它后面所有元素依此前移
代码如下:
时间复杂度: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(val == nums[i])
{
for(int j = i; j < size - 1; j++)
nums[j] = nums[j + 1];
i--;
size--;
}
return size;
}
};
也可以用双指针法,即用两个指针分别指向数组,fast指针判断如果不是要删除的元素,就把元素赋值给slow指针
代码如下:
时间复杂度: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];
slowIndex++;
}
}
return slowIndex;
}
};