前言:本人编写博文均为学习使用,若博文引用的内容存在侵权问题望告知,立即删改!
leetcode 15. 三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
话不多说,先上源码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<vector<int>> answer;
if(nums.size()<3||nums.front()>0||nums.back()<0) {
return answer;
}
int a,b,c;
int number=nums.size();
for(a=0;a<=number-3;a++) {
if(nums[a]>0)
break;
if(a>0&&nums[a]==nums[a-1]) {
continue;
}
b=a+1;
c=number-1;
while(b<c) {
if(nums[a]+nums[b]+nums[c]<0) {
b++;
}
else if(nums[a]+nums[b]+nums[c]>0) {
c--;
}
else {
answer.push_back({nums[a],nums[b],nums[c]});
while(b<c&&nums[b]==nums[b+1]) {
b++;
}
while(b<c&&nums[c-1]==nums[c]) {
c--;
}
b++;
c--;
}
}
}
return answer;
}
};
解题思路:
这道题目和leetcode 18 四数之和类似,我也是在做四数之和时,看了一位大神的题解写出来后,再来写三数之和的:
leetcode 18 四数之和 https://leetcode-cn.com/problems/4sum/
首先,很多人会想到采取暴力破解法来解决这个问题,但是在实际面试过程中,这样的解法必然不是满意的答案
这个解法首先将nums数组进行升序排序,保证最左边的数是最小的,右边的数是最大的,随后定义了3个指针,分别是a,b,c,分别指向三个数,其中a指针对应最左边的那个数,b指针对应中间那个数,c则指向最右边那个数,这三个数不断变化,指针在移动中,看能否找出答案,由于数组是升序排序的,所以,三个数肯定是最右边那个数最大,最左边那个数最小。在计算过程中,在a固定后,b,c两个指针进行移动,若三个数加起来大于0,则说明最右边,也就是第三个数太大了,将第三个数指针向左移,使三个数的和向0靠齐,若三个数加起来小于0,这说明三个数加起来太小了,优先考虑将第二个数的指针向右移动,向0靠齐。在不断移动中,找出最合适的解,并将结果存储起来。
几个注意的点:
1:由于计算的三个数的结果是0,所以,首先,若数组里面的数字个数小于3个,则肯定无解,返回空解,其次,若最左边的数大于0,那么中间的那个数和最右边的那个数更大于0,如此一来,便不能得到0这个结果了,所以,出现这种情况也返回空解,同理,若最右边的数小于0,也返回空解(因为在解题过程中,我们始终保持a,b,c对应的三个数是递增的)
2:有人可能会想到,为什么上述解法没有去重的操作,这是因为我始终是让a,b,c三个指针处于移动的状态,如上述代码所示,若nums[a]=nums[a-1],则跳过这一轮循环,同理,b,c也是如此,所以,这三个数始终都是变化的,来保证答案不会有重复的情况。