题目描述:
在数组中找出三个数字使得它们的和为0,找出所有的这种不包含重复的组合
思路:
首先将数组元素从小到大排序,设三个元素下标i,j,k,将i从头开始遍历,j初始化为i+1,k初始化为n-1;
while(j < k)
当三元素之和小于0时,j++;
当三元素之和大于0时,k–;
当三元素之和等于0时,记录此时的i,j,k;
while循环结束后执行i++,进行新一轮寻找
核心代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
vector<int> arr;
vector<int> temp; //存放所有和为0的组合
arr.push_back(2);
arr.push_back(3);
arr.push_back(1);
arr.push_back(0);
arr.push_back(-2);
arr.push_back(-1);
arr.push_back(-3);
arr.push_back(4);
int n = ();
sort((), ()); //默认从小到大排序
//排序后数组arr
for(int i = 0; i < n; i++)
cout << arr[i] << " ";
cout << endl;
for(int i = 0; i < n - 2; i++){
int j = i + 1;
int k = n - 1;
while(j < k){
if(arr[i] + arr[j] + arr[k] < 0)
j++;
else if(arr[i] + arr[j] + arr[k] > 0)
k--;
else{
temp.push_back(arr[i]);
temp.push_back(arr[j]);
temp.push_back(arr[k]);
j++;
k--;
}
}
}
for(int i = 0; i < (); i++){
cout << temp[i] << " ";
if(i % 3 == 2)
cout << endl;
}
return 0;
}