代码随想录算法训练营day1:
学习内容:
今天学习数组,有些题之前用C++刷过一遍了,今天试着用python写
学习产出:
二分查找:
二分法分两类
左闭右闭
class Solution:
def search(self, nums: List[int], target: int) -> int:
#标记起止点和中点
start,end=0,len(nums)-1#左闭右闭,所以取数组两个边界点
mid=(start+end)//2
while start<=end:#左闭右闭,所以边界可以相等,target如果存在最终会落到两个边界中的一点上或者中点
if nums[mid]<target:
start=mid+1#target不在这个闭区间内,所以要严格更新闭区间的边界
elif nums[mid]>target:
end=mid-1
else:
return mid
mid=(start+end)//2
return -1
左闭右开:
class Solution:
def search(self, nums: List[int], target: int) -> int:
#标记起止点和中点
start,end=0,len(nums)#左闭右开,所以取数组起始点和终止点外一点,因为右开取不到,这里很关键
mid=(start+end)//2
while start<end:#左闭右开,所以边界不可以相等,target如果存在最终会落到左边界或者区间中点
if nums[mid]<target:
start=mid+1#target不在开区间内,所以要严格更新闭区间的左边界
elif nums[mid]>target:
end=mid#右边界是开的,所以不用严格更新
else:
return mid
mid=(start+end)//2
return -1
移除元素:
用了快慢指针法原地修改,将等于val的元素都放到数组最后,比较好理解,有点像手动实现一个erase
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slow=0,fast=0;//定义快慢指针,快指针用来遍历,慢指针用来标记最后一个不等于val的元素的位置
int nums_vals=0;//定义等于val的元素数量
while(fast+nums_vals<nums.size()){//终止条件为快指针遍历完所有元素
if (nums[fast]==val){
for(int i=fast;i<nums.size()-1;i++){
nums[i]=nums[i+1];
}//快指针每识别一个等于val的元素就将之后的所有元素往左挪一位,
nums[end]=val;//然后把这个val放到数组最后;
nums_vals++;//val元素数量加一
}
else{
slow++;
fast++;//如果没有遍历到val,则继续遍历,慢指针也移动到最后一个不等于val的位置
}
}
return slow;//此时数组前【慢指针数】个元素都不等于val
}
};
用了库函数erase:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int count = 0;
int m = nums.size();
for(int i = 0; i < m; i++)
{
if(nums[i] == val)
{
nums.erase(nums.begin() + i);
i--;
m--;
}
}
return nums.size();
}
};
这个是官方的办法,更简洁:
class Solution {
public:
int removeElement(vector<int>& nums, int val) {
int slow=0,fast=0;
while(fast<nums.size()){
if (nums[fast]!=val){
nums[slow++]=nums[fast++];//每次遇到不等于val的元素都用slow标记
}
else{
fast++;//遇到等于val的话就跳过,直到遇到下一个不等于val的元素再更新slow
}
}
return slow;//slow此时标记完所有不等于val的元素
}
};