代码随想录算法训练营Day01
Leet code 704 二分查找 27 移除元素
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 中的所有元素是不重复的。
即使是二刷拿到题目的瞬间不想思考就是暴力法,一个循环遍历:
for(int i = 0;i<nums.length;i++){
if(nums[i]==target){
return i;
}
}
思考过后想起了二分查找法:
数组为有序数组,同时题目还强调数组中无重复元素,这是使用二分查找的前提条件。
二分查找法的重点是边界条件,有两种-左闭右闭,左闭右开
左闭右闭
public int search(int[] nums,int target){
if (target < nums[0] || target > nums[nums.length - 1]) {
return -1;
}
//左闭右闭
int left = 0;
int right = nums.length-1;
while(left<=right){
int middle = (left+right)/2;
if(target>nums[middle]){
left=middle+1;
}else if(target<nums[middle]){
right=middle-1;
}
else{
return middle;
}
}
return -1;
}
左闭右开
public int search(int[] nums,int target){
if (target < nums[0] || target > nums[nums.length - 1]) {
return -1;
}
//左闭右闭
int left = 0;
int right = nums.length; //注意这里边界范围
//左闭右开
while(left<right){
int middle = (left+right)/2;
if(target>nums[middle]){
left=middle+1;
}else if(target<nums[middle]){
right=middle;
}
else{
return middle;
}
}
return -1;
}
27.移除元素 题目描述
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例 1: 给定 nums = [3,2,2,3], val = 3, 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。 你不需要考虑数组中超出新长度后面的元素。
示例 2: 给定 nums = [0,1,2,2,3,0,4,2], val = 2, 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
你不需要考虑数组中超出新长度后面的元素。
因为之前做过一次,所以想起了快慢指针:
public int removeElement(int[] nums, int val) {
int fast = 0;
int slaw = 0;
while(fast<nums.length){
if(nums[fast]!=val){
nums[slaw++]=nums[fast];
}
fast++;
}
return slaw;
}
暴力解法
public static int removeElement(int[] nums, int val) {
int index = 0; // 用于记录新数组的索引位置
// 遍历原数组
for (int i = 0; i < nums.length; i++) {
// 当前元素不等于给定值时,将其放入新数组中
if (nums[i] != val) {
nums[index] = nums[i];
index++;
}
}
return index; // 返回新数组的长度
}
今日收获
由于之前数组部分已经做过,所以在拿到题目时不至于手足无措,但是在二分查找的时间左闭右开没有考虑好边界问题。