GEEK编程练习— —三数求和

Posted Sin_Geek

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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;
}

以上是关于GEEK编程练习— —三数求和的主要内容,如果未能解决你的问题,请参考以下文章

GEEK编程练习— —四数求和

GEEK编程练习— —雨水存储问题

GEEK编程练习— —格雷码

GEEK编程练习— —发糖果问题

数组练习题:两数之和三数之和四数之和

函数练习题-素数求和