代码随想录算法训练营第一天| 704. 二分查找、27. 移除元素。
704. 二分查找
工作以后好久没有刷了,看了一遍卡哥的两种解法。注意题目给定的条件:数组有序、不重复。
基本的思想:left和right通过while一直向内靠近,直到right>left,则不存在这样的数。然后自己做了一遍,没有好好审题,返回了要找到的数,结果题目要求下标。注意选择left<=right还是left<right的解法,也就是两种开闭区间。我选择的是第一种闭区间,因为left==right的情况是存在的。 (等过几天把第二种解法补充完整)
O(lgn). O(1)
// An highlighted block
int search(vector<int>& nums, int target) {
// 二分法:左闭右开的条件
int res = 0;
int left = 0;
int right = nums.size() - 1;
while(left <= right){
int mid = left + ((right - left) >> 1);
if (nums[mid] > target){
right = mid - 1;
} else if(nums[mid] < target){
left = mid + 1;
} else{
// res = nums[mid];
return mid;
}
}
return -1;
}
27. 移除元素
尝试暴力法解题:
O(n^2). O(1) 这个暴力法,一开始这个j边界条件写成了nums.size()没有通过,因为j的边界长度应该随size的变化而变化,想了一下原先的写法确实会越界的风险。
// An highlighted block
int removeElement(vector<int>& nums, int val) {
int size = nums.size();
for (int i = 0; i < size; i++)
{
/* code */
if (nums[i] == val){
for (int j = i + 1; j < size; j++)
{
/* code */
nums[j-1] = nums[j];
}
size--;
i--;
}
}
return size;
}
快慢指针法解题:
O(n). O(1). 快慢指针理解后还是比较好的。
int removeElement(vector<int>& nums, int val) {
int slow = 0;
// int fast = 0;
for (int fast = 0;fast < nums.size(); fast++ ){
if (nums[fast] != val){
nums[slow++] = nums[fast];
}
}
return slow;
}