计蒜客 难题题库 033 三个数的和

645次
6.20%
1000ms
65536K

给定数组S，判断数组中的元素知否存在a,b,c,使得a+b+c=0。
找出所有满足条件的元素并输出。
注意：元素a,b,c必须满足 a ≤ b ≤ c解集不重复。
格式：第一行输入一个数n，接下来一行输入数组S(n个元素)，接下来输出所有满足条件的解集。
如果不存在，则不输出任何东西。
例如：S = {-1 0 1 2 -1 -4}
则对应的解集为： (-1, 0, 1)和(-1, -1, 2)
提示：vector threeSum(vector &num)

样例1

输入：

6
3 0 -2 -1 1 2

输出：

-2 -1 3
-2 0 2
-1 0 1

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int main(){
int n, i, j, k;
vector<int> vi;
cin >> n;
vi.resize(n);
for(i = 0; i < n; ++i){
cin >> vi[i];
}
sort(vi.begin(), vi.end());
int temp;
for(k = 0; k + 2 < n; ++k){
while(k && k + 2 < n && vi[k - 1] == vi[k]){
++k;
}
i = k + 1;
j = n - 1;
while(i < j){
temp = vi[k] + vi[i] + vi[j];
if(temp == 0){
cout << vi[k] << " " << vi[i] << " " << vi[j] << " " << endl;
++i;
while(i < j && vi[i - 1] == vi[i]){
++i;
}
--j;
while(i < j && vi[j + 1] == vi[j]){
--j;
}
}else if(temp < 0){
++i;
}else{
--j;
}
}
}
}