第k小的值（题解）

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

题目描述

“O.o?”

集合中一开始拥有两个数字 a 和 b，如果 a 与 b 相同，那么仅有一个数字。

小沙每次可以选择集合中的两个数（可以相同）,将他们的和放入集合中，请问所有可能的集合中，第 K 小的值最小为多少？

注：集合中相同元素只能有一个。

输入描述:

第一行，输入三个整数 k，a，b 。

保证有 1≤k≤106，1≤a,b≤109。

输出描述:

输一个数字代表答案。

示例1

输入

3 2 3

输出

4

思路：

建立一个set，对于当前最小值x，我们有两种操作，要么当前值+a,要么当前值+b.循环k次，即为答案。

代码：

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<math.h>
#include<map>
using namespace std;
const int N =1e6+1000;
set<long long > p;
long long  n, a,b,cn;
void seek()
{
    while (1)
    {
       long long  x = *p.begin();
        p.erase(x);
        cn++;
        if (cn==n)
        {
            cout << x << endl;
            return;
        } 
        p.insert(x + a);
        p.insert(x + b);
    }
}
int main()
{
    cin >> n >> a >> b;
    p.insert(a);
    p.insert(b);
    seek();
    return 0;
}