当前位置: 代码迷 >> 综合 >> 在数组中找到和为0的全部组合
  详细解决方案

在数组中找到和为0的全部组合

热度:21   发布时间:2024-02-21 15:04:30.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()); //默认从小到大排序//排序后数组arrfor(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;
}

 

  相关解决方案