这个题目第一眼看上去就会想到用三重循环,但是用完之后会发现超时,所以需要去优化。
身为小白的我肯定是想不出来怎么优化的,于是我为了不浪费时间,果断去看了官方题解。
双指针:
当需要枚举数组中的两个元素的时候,如果发现这两个元素当其中一个是递增的时候,另一个随之递减,那么就可以将原本o(n^2)的时间复杂度降为o(n)。当左指针向右移动一个位置的时候,右指针会向左移动若干个位置。
在这个题目里面,因为有三个数,所以还需要在外面套一程循环,并且因为不能重复的原因,也就是这三个指针移动的时候,下一个元素都不能与上一个元素相同。
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& n) {
sort(n.begin(),n.end());//先排序
vector<vector<int>> ans;//存答案
for(int i=0;i<n.size();i++)//第一个元素
{
if(i!=0&&(n[i]==n[i-1])) continue;//(第一个元素没有上一个元素)如果当前元素与上一个元素相同,就直接跳过
int k=n.size()-1;//每次都更细第三个指针指向末尾
for(int j=i+1;j<n.size();j++)//第二个元素
{
if(j>i+1&&(n[j]==n[j-1])) continue;//原因同上
while(j<k&&(n[i]+n[j]+n[k]>0)) k--;//不断的循环,只要后面两个数的和大于第三个数,第三个指针就往左移,直到找到满足条件的数或者达到跳出这个循环的条件
if(k==j) break;
if(n[i]+n[j]+n[k]==0) ans.push_back({n[i],n[j],n[k]});
}
}
return ans;
}
};
(养成习惯)结尾附上官解代码:
https://leetcode-cn.com/problems/3sum/solution/san-shu-zhi-he-by-leetcode-solution/