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编程练习— —三数求和的主要内容,如果未能解决你的问题,请参考以下文章