Leetcode算法题 打卡day01
代码随想录算法训练营
二分查找
二分查找 - 查找target
排序数组。解决在一个排序数组中查找某个数或者寻找左侧边界或者右侧边界。
Leetocde 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
python
class Solution:
def search(self, nums: List[int], target: int) -> int:
left = 0
right = len(nums)-1
mid = (left+right)//2
while(left<=right): # 这个注意<=
if nums[mid] == target:
return mid
elif nums[mid] < target:
left = mid+1
elif nums[mid] > target:
right = mid-1
mid = (left+right)//2
return -1
c++
class Solution {
public:
int search(vector<int>& nums, int target) {
//
int left = 0;
int right = nums.size() - 1;
int mid = (left + right) / 2;
while(left<=right){
if (nums[mid] == target){
return mid;
}else if(nums[mid] > target){
right = mid-1;
}else if(nums[mid] < target){
left = mid+1;
}
mid = (left + right) / 2;
}
return -1;
}
};
二分查找 - 查找target or 插入值
Leetcode 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
python
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
left = 0
right = len(nums) -1
mid = (left+right)//2
flag = -1
while(left<=right):
if nums[mid] == target:
return mid
elif nums[mid] < target:
flag = 0
left = mid+1
elif nums[mid] > target:
flag = 1
right = mid-1
mid = (left+right)//2
return left
本题目和二分查找target基本一致,唯一不同是需要在目标值不存在于数组的时候返回插入的位置,即选择从未满足nums[mid]==target
时的返回值。
当目标值不存在于数组之中的时候,两个指针必然会先重合(设此处的值是x
)之后再移动一次后退出while
循环。此处分类讨论,当x
大于目标值的是,右侧指针会移动right=mid-1
,此时目标值需要插在x
的前面,于是返回left
,当x
小于目标值的时候,left=left+1
,此时应当将值插在x
后方,综上两种情况都是返回left
。
移除元素
Leetcode 27 给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素。元素的顺序可能发生改变。然后返回 nums 中与 val 不同的元素的数量。
假设 nums 中不等于 val 的元素数量为 k,要通过此题,您需要执行以下操作:
更改 nums 数组,使 nums 的前 k 个元素包含不等于 val 的元素。nums 的其余元素和 nums 的大小并不重要。
返回 k。
python
class Solution:
def removeElement(self, nums: List[int], val: int) -> int:
slow = 0
fast = 0
while(fast<len(nums)):
if nums[fast] != val:
nums[fast], nums[slow] = nums[slow], nums[fast] # 交换
# 也可以直接覆盖
# nums[slow] = nums[fast]
fast += 1
slow += 1
else:
fast+=1
return slow
移除元素 == 将不是这个元素的数据向前移动。使用双指针 fast
和slow
两个指针,其中slow
指针负责寻找位置 ,fast
指针负责寻找不等于val的元素。