题目
题解
这题还是比较简单的,首先进行排序(对于没有按下标输出要求的题目,向考虑排序),排好序后,找第一个数,设第一个数下标为i,则第二个数从i+1开始找,因为i之前的数字的所有组合都已经找出来了,计算第一个数与第二数的和,取反,根据二分法查找,查找范围为第二个数的下标到数组结尾,另外当第一个数为正数时就可以退出了,因为正数后面都为正数,不可能为0,另外题目要求不能重复,一些边界条件还是要注意
代码
class Solution {
public:
int binary(vector<int>&a,int first,int end,int number) {
int mid;
while (first <= end) {
mid = (first + end) / 2;
if (a[mid] == number)
return mid;
if (a[mid] < number)
first = mid + 1;
else
end = mid-1;
}
return -1;
}
vector<vector<int>> threeSum(vector<int>& a) {
sort(a.begin(), a.end());
vector<vector<int>> result0;
int i,j,n=a.size();
for (i = 0;i < n;i++) {
if (a[i] > 0) break;
if (i == 0 || a[i] != a[i - 1]) {
for (j = i + 1;j < n;j++) {
int temp = 0 - a[i] - a[j];
int result = binary(a, j + 1, n - 1, temp);
if ((j==i+1||a[j]!=a[j-1])&&result != -1) {
vector<int> temp;
temp.push_back(a[i]);
temp.push_back(a[j]);
temp.push_back(a[result]);
result0.push_back(temp);
}
}
}
}
return result0;
}
};