GEEK编程练习— —三数求和

题目

输入一个特定整数s和一组整数，要求从这组整数中找到三个数a，b，c，使a+b+c=s。按照升序排列，输出所有满足条件的a，b，c。具体格式如下：

输入

 0
 -1 0 1 2 -1 -4

输出

-1 -1 2
-1 0 1

分析

先排序，然后左右夹逼，复杂度O(n^2)
此方法可以推广到k个数求和，先排序，做k-2次循环，在最内层循环左右夹逼，时间复杂度O(max(nlogn，n^(k-1)))

代码

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main()
{
    vector<int> nums;
    int target;
    int tmp;
    cin >> target;
    while (cin >> tmp)
        nums.push_back(tmp);

    vector<int> result;
    if (nums.size() < 3)
        return 0;
    sort(nums.begin(),nums.end());

    auto last = nums.end();
    for (auto i = nums.begin(); i < last - 2; ++i)
    {
        auto j = i + 1;
        //跳过重复数
        if (i > nums.begin() && *i == *(i - 1))
            continue;
        auto k = last - 1;
        while (j < k)
        {
            if (*i + *j + *k < target)
            {
                ++j;
                //跳过重复数
                while (*j == *(j - 1) && j < k)
                    ++j;
            }
            else if (*i + *j + *k > target)
            {
                --k;
                //跳过重复数
                while (*k == *(k + 1) && j < k)
                    --k;
            }
            else
            {
                result.push_back(*i);
                result.push_back(*j);
                result.push_back(*k);
                ++j;
                --k;
                //跳过重复数
                while (*k == *(k + 1) && *j == *(j - 1) && j < k)
                    ++j;
            }
        }
    }

    for (auto i = 0; i < result.size(); i += 3)
    {
        cout << result[i]<< ends << result[i + 1] << ends << result[i + 2] << endl;
    }
    return 0;
}