三数之和
这种问题第一个想到的就是穷举法,可是这样必然超时,于是我想用上一道题 two sum的方法,定两个变量找一个,尝试之后失败了,定的两个变量没有办法有序的变化。
于是用双指针求和,就是定一个,用双指针代表两个变量,按照要求移动。其实还是之前的方法。以后求两数和的时候都可以用双指针(前提是排好序),下面上代码
class Solution {
public:
/*
* @param numbers: Give an array numbers of n integer
* @return: Find all unique triplets in the array which gives the sum of zero.
*/
vector<vector<int>> threeSum(vector<int> &nums) {
// write your code here
vector<vector<int>>a;
sort(nums.begin(),nums.end());//先进行排序
for(int i=0;i<nums.size();i++)//思路是双指针,定一找二
{
if(nums[i]>0)//最小的数字已经大于0就不用找了
break;
if(nums[i-1]==nums[i]&&i>0)//这里不能写成if(nums[i]==nums[i+1])这样可能把满足条件的数字跳过去,如
//[8,8,-16],所以当相邻数相同时,当一个已经算过后再跳
continue;
int k=i+1,j=nums.size()-1;
while(k<j)//双指针的算法要时刻记得判断两指针位置!!
{
if(nums[k]+nums[j]==-nums[i])//满足条件时,两个指针一块走
{ a.push_back({nums[i],nums[k],nums[j]});//要返回vector<vector<int>>形式,可以这样写vector<int>,即加大括号{}
while(k<j&&nums[k]==nums[k+1])k++;
while(k<j&&nums[j]==nums[j-1])j--;
k++;j--;
}
else if(nums[k]+nums[j]<-nums[i])//小于所求的数时,左边指针往右走
k++;
else//大于所求的数时,右边指针往右走
j--;
}
}
return a;
}
};