代码随想录算法训练营第一天|704.二分查找、27.移除元素、35.搜索插入位置
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
思路:就是比较正常的二分查找。选一个中间点,然后将目标值和数组的这个中间值比较,如果目标值大于中间值,说明查找的结果在右边那部分,那么左边指针=mid+1,反之,右边指针=mid-1。如果刚好等于中间值,那么返回中间指针。
代码
class Solution {
public int search(int[] nums, int target) {
int l=0,r=nums.length-1;
while(l<=r){
int mid=(l+r)/2;
if(target==nums[mid]){
return mid;
}
if(target<nums[mid]){
r=mid-1;
}else {
l=mid+1;
}
}
return -1;
}
}
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,需要注意的是慢指针不受快指针移动的影响,不要快指针移动就想着慢指针也要动。用一个for循环,实现fast指针的遍历,当指向的值不是val时,那么将这个值给慢指针,并对慢指针进行++操作,如果这个值等于val,那么fast继续走,slow不走。也就是说,fast一直在不受影响的往前走,slow只有当fast不等于目标值才走。最后数组的长度就是slow的值,因为slow最后一步还进行了一个++操作。
代码
class Solution {
public int removeElement(int[] nums, int val) {
int slow=0;
for(int fast=0;fast<nums.length;fast++){
if(nums[fast]!=val){
nums[slow++]=nums[fast];
}
}
return slow;
}
}
暴力解法(会超时)
**思路:**两层for循环。第一层for循环正常遍历,当遇到等于val 的值时,进行第二次for循环:从等于的这个值开始,后面的值一次覆盖前一个值,也就是往前推进一个位置。
代码
class Solution {
public int removeElement(int[] nums, int val) {
int size=nums.length;
for(int i=0;i<nums.length;i++){
if(nums[i]==val){
for(int j=i+1;j<nums.length;j++){
nums[j-1]=nums[j];
}
}
i--;
size--;
}
return size;
}
}
35、搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n)
的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
题解:二分查找目标值。如果找到则返回下标,未找到,最终返回左下标。
为什么最后返回的是做下标呢?
因为如果没有找到目标值,那么最后退出的条件是 左边大于右边,也就是 left>right,但是数组是按大小排序的,目标值要插入的位置是较大的那个数的位置,然后较大的那个数要往后移。*
代码
class Solution {
public int searchInsert(int[] nums, int target) {
int l=0,r=nums.length-1;
while(l<=r){
int mid=(l+r)/2;
if(nums[mid]==target){
return mid;
}
if(nums[mid]<target){
l=mid+1;
}else{
r=mid-1;
}
}
return l;
}
}