在数组中找到和为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 = arr.size();
	
	sort(arr.begin(), arr.end()); //默认从小到大排序
	//排序后数组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 < temp.size(); i++){
		cout << temp[i] << " ";
		if(i % 3 == 2)
			cout << endl;
	}
	
	return 0;
}

 

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页