双指针-OJ题
- 移动零(点击跳转)
- 原理讲解
- 代码实现
- 复写零(点击跳转)
- 原理讲解
- 代码实现
- 快乐数(点击跳转)
- 原理讲解
- 代码实现
- 盛最多水的容器(点击跳转)
- 原理讲解
- 代码实现
- 有效三角形的个数(点击跳转)
- 原理讲解
- 代码实现
- 查找总价值为目标值的两个商品(点击跳转)
- 原理讲解
- 代码实现
- 三数之和(点击跳转)
- 原理讲解
- 代码实现
- 四数之和(点击跳转)
- 原理讲解
- 代码实现
移动零(点击跳转)
原理讲解
因此我们定义两个指针:
- cur:遍历数组
- dest:已处理的区间内,最后一个非零元素的位置
这样就把数组划分成三块区间:
[0 , dest] 、[dest+1 , cur-1] 、[cur , size-1]
分别对应:
非零元素、零、待处理
具体操作:
cur遍历过程中
- 遇到0元素
cur++; - 遇到非0元素
swap(dest+1,cur); //swap数组中下标为dest+1和cur的元素
dest++;cur++;
代码实现
void moveZeroes(vector<int>& nums) {
int dest = -1;
int cur = 0;
while (cur < nums.size())
{
if (nums[cur])
swap(nums[++dest], nums[cur]);
cur++;
}
}
复写零(点击跳转)
原理讲解
这道题首先要想到从后向前遍历,因为如果从前向后遍历,会覆盖掉后面的值,较难操作,因此:
第一步找到最后一个数,可以用快慢指针来实现:cur指针、dest指针。前者从前向后遍历,后者根据前者位置的值走1步或2步,具体如下:
- 先判断cur位置的值
- 若cur位置的值非0,dest向后移动一步;若cur位置的值为0,dest向后移动2步
- 判断dest位置是否到了结束的位置
- cur++
代码实现
void duplicateZeros(vector<int>& arr) {
int dest = -1;
int cur = 0;
//找到最后一个数
while (cur < arr.size())
{
if (arr[cur])
dest++;
else
dest += 2;
if (dest >= arr.size() - 1)
break;
cur++;
}
//处理特殊情况:cur指向的最后一个数是0,dest越界
if (dest == arr.size())
{
arr[dest - 1] = 0;
cur--;
dest -= 2;
}
//复写
while (cur >= 0)
{
if (arr[cur])
arr[dest--] = arr[cur--];
else
{
arr[dest--] = 0;
arr[dest--] = 0;
cur--;
}
}
}
快乐数(点击跳转)
原理讲解
最后一定会成环(可以证明,用鸽巢原理)
因此可以通过判断环的起点是否为1,决定返回true
还是false
→ 快慢指针
快慢指针有⼀个特性,就是在⼀个圆圈中,快指针总是会追上慢指针的,也就是说他们总会相遇在⼀个位置上。就本题而言,如果相遇位置的值是 1 ,那么这个数⼀定是快乐数;如果相遇位置不是 1 ,那么就不是快乐数。
代码实现
//求x每一位数的平方和
int SqSum(int x)
{
int ret = 0;
while(x)
{
int n = x%10;
ret += n*n;
x/=10;
}
return ret;
}
bool isHappy(int n) {
int slow = n,fast = SqSum(n);//fast不能设成n,否则进不了循环
while(slow != fast)
{
slow = SqSum(slow);
fast = SqSum(SqSum(fast));
}
return slow==1;
}
盛最多水的容器(点击跳转)
原理讲解
我们首先想到的大概率是两层循环暴力枚举,但是复杂度高,这道题不能通关
那应该如何解这题呢?
首先,容器的容积(这道题只考虑面积)是S = h * w
h表示高度,即height[?]
w表示宽度,即两条垂线间隔的距离
为了方便叙述,我们假设左边边界height[left]
小于右边边界height[right]
。
如果此时我们固定⼀个边界,改变另⼀个边界,水的容积会有如下变化形式:
- 容器的宽度w一定变小。
- 由于左边界较小,决定了水的高度。①如果改变左边界,新的水面高度不确定,但是一定不会超过右边的柱子高度,因此容器的容积可能会改变(可能变大、变小、不变)。
- ②如果改变右边界,无论右边界移动到哪里,新的水面的高度一定不会超过左边界,也就是不会超过现在的水面高度,但由于容器的宽度减小,因此容器的容积⼀定会变小的。
所以我们可以舍去情况②,只需要讨论情况①(因为我们的目的是求最大的容积)
因此,我们定义两个指针left和right,然后比较height[left]
和 height[right]
,移动height小的位置的指针,循环这个过程,期间产生的所有的容积里的最大值,就是要return的最终答案
代码实现
int maxArea(vector<int>& height) {
int left = 0;
int right = height.size()-1;
int _max = 0;
while(left < right)
{
int tmp = (right-left)*min(height[right],height[left]);
_max = max(_max,tmp);
if(height[left] < height[right])
left++;
else
right--;
}
return _max;
}
有效三角形的个数(点击跳转)
原理讲解
我们小学五年级都学过,三角形的三条边必须满足:任意两边之和大于第三边、任意两边之差小于第三边~
假设三角形三条边长度分别为:a, b, c
-
方法① a+b < c ; a+c < b ; b+c < a
-
方法② 在①的基础上优化:先排序,a ≤ b ≤ c,只需要判断 a+b > c即可
解法一:三层循环暴力枚举 → O(n3)
解法二:利用(排序后)单调性,用双指针算法来解决。 → O(n2)
- 先固定最大的数
- 在最大数的左边区间内,用双指针算法,快速统计出符合要求的另外两个数:
-
- 如果
nums[left]
+nums[right]
>nums[max]
,说明 [left, right - 1] 区间上的所有元素均可以与nums[right]
构成比nums[max]
大的二元组,此时满足条件的有 right - left 种;此时 right 位置的元素的所有情况相当于全部考虑完毕, right-- ,进入下一轮判断
- 如果
-
- 如果
nums[left]
+nums[right]
<=nums[max]
,说明 left 位置的元素是不可能与 [left + 1, right] 位置上的任意元素构成满足条件的二元组,left++
进入下一轮循环
- 如果
- 向左移动最大数,循环上面的过程
代码实现
int triangleNumber(vector<int>& nums) {
int left,right;
int cmax = nums.size() - 1;
int ret = 0;
sort(nums.begin(),nums.end());
while (cmax > 1) {
left = 0;
right = cmax-1;
while (left < right) {
if (nums[left] + nums[right] > nums[cmax]) {
ret+=(right-left);
right--;
} else {
left++;
}
}
cmax--;//向左移动最大数
}
return ret;
}
查找总价值为目标值的两个商品(点击跳转)
原理讲解
类比上面《有效三角形的个数》,会发现利用单调性同样很香:利用对撞指针优化时间复杂度
- 初始化
left
,right
分别指向数组的左右两端 - 接下来无非三种情况:
-
price[left]
+price[right]
==target
,说明找到了,跳出循环返回即可
-
price[left]
+price[right]
>target
,说明两数之和大了,right–
-
price[left]
+price[right]
<target
,说明两数之和小了,left++
代码实现
vector<int> twoSum(vector<int>& price, int target) {
int left = 0,right = price.size()-1;
vector<int> ret;
while(left < right)
{
if(price[left] + price[right] == target)
{
ret.push_back(price[left]);
ret.push_back(price[right]);
break;
}
else if(price[left] + price[right] > target)
right--;
else
left++;
}
return ret;
}
或者不需创建vector直接返回:
vector<int> twoSum(vector<int>& price, int target) {
int left = 0,right = price.size()-1;
while(left < right)
{
if(price[left] + price[right] == target)
return {price[left] , price[right]};
else if(price[left] + price[right] > target)
right--;
else
left++;
}
return {-1};//注意这里必须return一个数组,目的是照顾编译器,否则编译不通过
}
三数之和(点击跳转)
原理讲解
我们可以利用在两数之和那道题的双指针思想,来对暴力枚举做优化:
- 先排序
- 然后固定一个数 a
- 在这个数后面的区间内,利用双指针快速找到两个数之和等于 -a 即可
但是要注意的是,这道题需要有去重操作~
(除了用set,我们可以自己实现)
- 找到一个结果之后, left 和 right 指针要跳过重复的元素;
- 当用完⼀次双指针算法之后,固定的 a 也要跳过重复的元素。
代码实现
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
int i =0,left,right;
vector<vector<int>> vv;
while(i < nums.size()-2)
{
if(nums[i] > 0)
break;
//left+right
left = i+1;
right = nums.size()-1;
while(left < right && right < nums.size())
{
if(nums[left] + nums[right] == -nums[i])
{
vv.push_back({nums[i],nums[left],nums[right]});
left++;right--;
while(left < right && right < nums.size()&&nums[left] == nums[left-1])
left++;//left跳过重复元素
while(left < right && right < nums.size()&&nums[right] == nums[right+1])
right--;//right跳过重复元素
}
else if(nums[left] + nums[right] > -nums[i])
right--;
else
left++;
}
i++;
while(nums[i] == nums[i-1] && i<nums.size()-2)
++i;//“固定”的数跳过重复元素
}
return vv;
}
四数之和(点击跳转)
原理讲解
这道题和上面的《三数之和》几乎一模一样,区别在于多了一层
- 固定一个数a
- 在这个数 a 的后面区间上,利用「三数之和」找到三个数,使这三个数的和等于
target - a
即可
代码实现
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(),nums.end());
vector<vector<int>> vv;
int i=0,j,left,right;
while(i<nums.size())
{
j = i+1;
while(j<nums.size())
{
left = j+1;
right = nums.size()-1;
long long aim = (long long)target-nums[i]-nums[j];//有些用例不强转会越界
while(left<right)
{
int sum =nums[left]+nums[right];
if(sum == aim)
{
vv.push_back({nums[i],nums[j],nums[left],nums[right]});
left++;right--;
while(left<right && nums[left] == nums[left-1])
left++;
while(left<right && nums[right] == nums[right+1])
right--;
}
else if(sum < aim)
left++;
else
right--;
}
++j;
while(j < nums.size() && nums[j] == nums[j-1])
++j;
}
++i;
while(i<nums.size() && nums[i] == nums[i-1])
++i;
}
return vv;
}