代码随想录刷题第一天|704二分查找 27移除元素
704 二分查找
**题目:**给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
思路: 因为题目中的数组有序,且没有重复元素,可以使用二分查找找到唯一下标。二分查找的两种写法
1. left < right
target在一个左开右闭的[left,right)的区间内,判断条件使用while (left < right),这里使用 < ,因为left == right在区间[left, right)是没有意义的,right 更新为 middle,因为当前nums[mid]不等于target,去左区间继续寻找,而寻找区间是左闭右开区间,所以right更新为mid,即:下一个查询区间不会去比较nums[mid]
代码:
class Solution {
public int search(int[] nums, int target) {
if(target < nums[0] || target > nums[nums.length - 1]){
return -1;
}
int left = 0, right = nums.length;
while(left < right){
int mid = (left + right) / 2;
if(nums[mid] == target){
return mid;
}else if(nums[mid] > target){
right = mid;
}else{
left = mid + 1;
}
}
return -1;
}
}
2. left <= right
target在一个左开右开的[left,right]的区间内,判断条件使用while(left<=right),因为==时有意义,if (nums[mid] > target) right 要赋值为 mid - 1,因为当前这个nums[mid]一定不是target,那么接下来要查找的左区间结束下标位置就是 mid - 1
代码:
class Solution {
public int search(int[] nums, int target) {
if(target < nums[0] || target > nums[nums.length - 1]){
return -1;
}
int left = 0, right = nums.length - 1;
while(left <= right){
int mid = (left + right) / 2;
if(nums[mid] == target){
return mid;
}else if(nums[mid] > target){
right = mid - 1;
}else{
left = mid + 1;
}
}
return -1;
}
}
时间复杂度O(log2n),空间复杂度O(1)
27 移除元素
题目: 给你一个数组 nums
和一个值 val
,你需要 原地 移除所有数值等于 val
的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1)
额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路: 采用快慢指针的方法,快指针遍历数组,慢指针指向待插入的位置(图片来源:代码随想录)
代码:
class Solution {
public int removeElement(int[] nums, int val) {
int low = 0, fast = 0;
while(fast < nums.length){
if(nums[fast] != val){
nums[low] = nums[fast];
low++;
}
fast++;
}
return low;
}
}
时间复杂度O(n),空间复杂度O(1)