2021年4.19TX笔试题第二题题解

发广告

将k条新的广告按照用户设定的时间顺序发放到用户的邮箱。
如果某用户设定的时间为t,那么这位用户接收第一封广告的时间是t，接收第二封广告的时间是2*t，以此类推。

输入描述：

第一行输入两个正整数，分别是用户数量和广告个数k。
第2到n+1行，每行一个正整数，代表编号为i的用户，设定的时间t。

输出描述：

输出一共k行，按照发放广告的时间顺序，依次输出用户的编号。如果某一时间，有多个用户可以接收广告，优先给用户编号小的发放。

输入示例：

3 3 
3
6
9

输出示例：

1
1
2

思路：

提炼内容，就是接收广告时间最小的用户收到第一封信件，如果有多个时间最小的用户，选择用户id最小的用户。
利用优先队列来完成。
优先队列中存储，广告时间点和用户id，利用pair。
每次找到接收用户后，弹出队列，将时间增加，再塞入队列。

代码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
struct cmp
{
    bool operator() (pair<int, int>& a, pair<int, int>& b)
    {
        return (a.first > b.first) || (a.first == b.first) && (a.second > b.second);
    }
};
int main()
{
    int n, k;
    cin >> n >> k;
    vector<int> times(n);
    priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> prior;
    int time;
    for (int i = 0; i < n; i++)
    {
        cin >> time;
        times[i] = time;
        prior.push(make_pair(time, i));
    }
    for (int i = 0; i < k; i++)
    {
        int j = prior.top().second;
        time = prior.top().first + times[j];
        cout << j + 1 << endl;
        prior.pop();
        prior.push(make_pair(time, j)); 
    }
    return 0;
}

如有错误之处，还请不吝赐教。