【力扣15 三数之和 指针法详解】

题目描述

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意:答案中不可以包含重复的三元组。

示例

示例 1:

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:

输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:

输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。

个人想法

本道题看似简单但是在编写代码时还是会遇到一些问题需要小心:
①去重逻辑(本题的关键)
②指针移动逻辑
③要区分if连用与if 和else if分开用的区别(是个简单的语法问题,但是本人长时间没写代码最开始没注意报错才反应过来 让大家见笑了)

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>0)
            {
                return result;
            }
            if(i>0 && nums[i]==nums[i-1])
            {
                continue;
            }
            int right=nums.size()-1;
            int left=i+1;
            while(right>left)
            {
                if(nums[i]+nums[right]+nums[left]==0)
                {
                    result.push_back(vector<int>{nums[i],nums[left],nums[right]});
                    while(left < right && nums[left]==nums[left+1]) left++;
                    while(right > left && nums[right]==nums[right-1]) right--;
                    left++;
                    right--;
                }
                else if(nums[i]+nums[right]+nums[left]>0) {right--;}

                else if(nums[i]+nums[right]+nums[left]<0) {left++;}

            }
        }
        return result;
    }
};

这是本人自己根据题目解法思路写的代码,不怎么规范,想看规范的可以参考官方的代码链接: 代码随想录

思路

①先将元素排序
②确定第一个元素(要去重)
③在剩余元素中寻找符合的另外两个元素(因为做了排序,只要在第一个元素右边寻找;第三步就是双指针法;寻找的后两个元素也要去重)

代码详解

vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;

这里两行代码是函数的定义,定义了解决这个问题的函数名字,返回值,参数。
vector<vector>表明返回值是一个容器,每个容器元素又是一个容器,简单理解就是返回一个二维数组,这个数组的大小可以改变,与传统数组有区别叫容器。这个解释清楚,那这两行代码就没什么难理解的地方了。

sort(nums.begin(),nums.end());

将nums容器里面的元素进行排序,sort是c++自带的函数。
begin和end是容器自带的函数,分别返回第一个元素的指针与最后一个元素下一个位置的指针。

if(nums[i]>0)
{
	return result;
}

当nums排序好后第一个元素大于0也就不存在相加等于零的情况,而且以后也不会再有满足该条件的情况了 所以直接返回结束函数。

 if(i>0 && nums[i]==nums[i-1]) 
 {
 	continue;
}

第一个元素的去重语句,判断条件不可以是下面这个条件:

if (nums[i] == nums[i + 1]) 

因为如果-1,-1,2这种情况也会被该判断语句跳过
所以改进一下,但是第一个元素在判断这个i-1时会越界,所以加个i>0的判断,并且&&两边的顺序不能调换,因为根据语法执行顺序如果调换也是会报错的。

int right=nums.size()-1;
int left=i+1;

确定好第一个元素位置了,开始初始话左右指针(这里就是题目说的双指针的两个指针,但是我个人理解因为有i的缘故 觉得叫三指针也没问题,所以我在题目上只说指针解法,有的课本好像也叫哨兵)

while(right>left)
{
	if(nums[i]+nums[right]+nums[left]==0)
 	{
		result.push_back(vector<int>{nums[i],nums[left],nums[right]});
        while(left < right && nums[left]==nums[left+1]) left++;
        while(right > left && nums[right]==nums[right-1]) right--;
        left++;
        right--;
    }
    else if(nums[i]+nums[right]+nums[left]>0) {right--;}

    else if(nums[i]+nums[right]+nums[left]<0) {left++;}

}

循环条件很好理解,这里核心主要是第一个if里面的语句,找到满足条件的三元组,将它送到result里面,接下来就是过滤掉左右指针相邻的重复的元素,因为排序过以后重复的元素一定是挨在一起的。
过滤完重复的元素,左右指针一起收缩。
另外两个条件也好理解,当相加的元素大于0,说明比要求偏大,把偏大的那边往小移动,当相加元素小于零,把偏小的元素向变大的方向移动,这些都是建立在有序的前提下。

  • 7
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值