本文两个题目与解题思路均来自代码随想录-数组章节,加入了一些自己的内容与思考。
1. leetcode 704.(二分)查找
问题简述
给定一个n个元素有序的(升序)整型数组nums和一个目标值target,写一个函数搜索nums中的target,如果目标值存在返回下标,否则返回-1。
思路描述
-
本题属于算法中较为基础的问题——查找问题(《算法导论(原书第三版)》练习2.1-3),对于这一问题,最直观的想法就是使用for循环遍历数组,将数组中的每个元素与target进行比较,如果相等则输出相应位置的下标,如果遍历过程没有发现相等的值,则查找失败,返回-1。这种算法的时间复杂度为 O ( n ) O(n) O(n)。
-
二分查找是一种低复杂度的查找算法,时间复杂度为 O ( l o g n ) O(logn) O(logn),但对数组有以下要求
-
数组元素有序(否则不能实现每次循环将查找的范围一分为二)
-
数组中无重复元素(否则二分法找到的下标值不唯一,不能输出所有与target相等的元素)
实现二分查找的方法为设置左右指针left和right,构成一个区间[left, right]或[left, right),在while循环中,区间作为循环不变量,始终代表“待查找的子数组”。
-
算法实现
二分查找算法的C语言实现:
int search(int* nums, int numsSize, int target){
int left = 0;
int right = numsSize - 1;
int middle = 0;
while(left <= right){
middle = left + ((right - left) >> 1);
if(target > nums[middle]){
left = middle + 1;
}
else if(target < nums[middle]){
right = middle - 1;
}
else{
return middle;
}
}
return -1;
}
核心要点
- (不变量)区间初始化,while循环终止的条件,以及左右指针的更新
以闭区间的情况[left, right]为例(事实上,闭区间与开区间的区别便是端点是否在区间内)。
-
(初始化)由于区间是闭的,此时left和right分别指向数组的第一个和最后一个元素(也就是说这两个元素包含在待查找的区间内),即
left = 0
,right = numSize - 1
。 -
(循环终止条件)由于区间是闭的,因此
left = right
是合法的,因此while循环的终止条件应该为left <= right
。 -
(左右指针更新)若
target > nums[middle]
成立,我们知道middle下标对应的元素不在我们下次的搜索范围(区间)内,由于区间是闭的,区间的端点在区间范围,因此我们在更新left的时候,应当使left = middle + 1
,从而区间不包含middle对应的元素。(right的更新同理)
附:如果是左闭右开区间,我们直接给出结论
- 初始化时,
right = numSize
- 循环终止条件为
left < right
- 此时左右指针的更新不同,更新时
left = middle + 1
,而right = middle
。
- (细节)两个整型相加可能出现的溢出问题
在计算中点下标middle时,有两种在数学上等价的计算思路
middle = (left + right) / 2
middle = left + (right - left) / 2
但第一种方式中包含了两个整型数相加,当两个整型数都很大时,可能会发生溢出,因此用第二种计算方法更为保险。
第二种计算方法也可以用右移位运算符>>
实现,将一个整数的二进制形式向右移动一位,就相当于除以2。
2. leetcoede 27.移除元素
问题简述
给你一个数组nums和一个值val,你需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O ( 1 ) O(1) O(1)额外空间并原地修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
思路描述
- 本题属于数组的一种基本操作的实现。与上一题类似,本题一个直观的想法就是利用for循环遍历数组,每当发现一个需要删除的元素,就再次利用for循环使这个元素之后的所有元素依次前移,从而覆盖我们想要删除的值。这种算法的时间复杂度为 O ( n 2 ) O(n^2) O(n2)。
- 利用快慢指针实现元素的移除只需要 O ( n ) O(n) O(n)的时间复杂度。这种方法通过设置快慢指针fast和slow,让fast指针遍历数组,当发现数组中需要保留的数时,将这个数赋给slow指针指向的位置,slow指针向后移一位,等待fast指针再次发现需要保留的数。当fast指针遍历完成后,我们就在原地构建了一个新数组,数组中不含需要删除的数。
算法实现
利用快慢指针移除元素算法的C语言实现:
int removeElement(int* nums, int numsSize, int val){
int slow = 0;
for(int fast = 0; fast < numsSize; fast++){
if(nums[fast] != val){
nums[slow++] = nums[fast];
}
}
return slow;
}
核心要点
- 快慢指针的使用
如果题目条件中没有“原地”的限制,本题可以创建一个新数组,通过for循环遍历原数组,在这段新的存储空间上“有选择地复制”原数组的部分元素。因此,本题使用的慢指针本质上就相当于在原数组的内存空间上进行这种“有选择的复制”。
“输出链表的中间节点”,“链表中倒数第k个节点”等问题,也可以用快慢指针解决。
- (细节)返回值
本题要求返回删除元素后数组的长度,此长度为数组中最后一个元素下标加1。在算法实现中,当nums[fast]
对nums[slow]
最后一次赋值后,由于slow++
,slow指针恰好指向新数组最后一个元素的下一个位置,因此返回slow即可。