代码随想录算法训练营第一天 | LeetCode704.二分查找、LeetCode27.移除元素
前言
今天的内容主要为(1.对数组进行操作),(2.二分查找)
ps:本人第一次写博客,有处理不当的地方请见谅!!!
一、704.二分查找
题目描述:
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
示例 1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
示例 2:
输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
另外还有提示如下:
你可以假设 nums 中的所有元素是不重复的。
n 将在 [1, 10000]之间。
nums 的每个元素都将在 [-9999, 9999]之间。
解题思路:
给定一个有序的数组,给定一个目标值,然后搜索这个目标值的下标,题意很清楚明了。如果暴力来做,时间复杂度为O(n),如果用二分查找来做,时间复杂度为O(logn)。
解题代码C++:
class Solution {
public:
int search(vector<int>& nums, int target) {
int l = 0, r = nums.size() - 1;
while(l < r)
{
int mid = l + r >> 1;
if(nums[mid] >= target) r = mid;
else l = mid + 1;
//或者
//int mid = l + r + 1 >> 1;
//if(nums[mid] <= target) l = mid;
//else r = mid - 1;
}
if(nums[l] == target) return l;
else return -1;
}
};
注意点:
我使用的二分模板为acwing-yxc的模板,使用这种二分模板的时候需要注意,中间值mid的设置是否需要+1,这个是非常关键的注意点,搞错会tle
二、27.移除元素
题目描述:
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1:
输入:nums = [3,2,2,3], val = 3
输出:2, nums = [2,2]
解释:函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如,函数返回的新长度为 2 ,而 nums = [2,2,3,3] 或 nums = [2,2,0,0],也会被视作正确答案。
示例 2:
输入:nums = [0,1,2,2,3,0,4,2], val = 2
输出:5, nums = [0,1,3,0,4]
解释:函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。
另外还有提示如下:
0 <= nums.length <= 100
0 <= nums[i] <= 50
0 <= val <= 100
解题思路:
因为我们需要在原地修改数组,而不使用额外的数组空间,首先我们要想到只能对这个数组,在这个数组内部操作。这个时候会想到堆操作的例子,在堆里面我们要删除一个数,会把这个数跟堆头交换,然后删除堆头,达到O(1)的时间复杂度。而在这道题,换汤不换药,我们可以对数组进行操作,找到一个目标位置的元素,这里我们就选择数组的最后一个元素,然后从后往前遍历数组,如果遇到元素值等于val,我们把这个元素与最后一个元素交换值,然后把数组长度减小一,反复行此。
解题代码C++:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int t = nums.size();
for(int i = t - 1; i >= 0; i --)
{
if(nums[i] == val)
{
swap(nums[i], nums[t - 1]);
t --;
}
}
return t;
}
};