数据结构综合应用——堆

洛谷1631

题意：给两个有序序列a, b，长度都是N，在每个序列中各取一个数相加，则有N^2个数，输出最小的前N个数

思路：
无论如何，a序列的第一位数一定要用到，先将这个数和b中每一个数相加，标记，并扔进大根堆中排序，每次输出一个最小值，追溯这个值来源的下标，并根据下标插入新的值*
其中a[t[m]] + b[m]，t[m]所存的的值表示a序列加到哪一位数了，m表示b序列加到哪一位数了

#include<bits/stdc++.h>
using namespace std;

int a[100010], b[100010];
int t[100010];
priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > qe;

int main() {
    int n; cin >> n;
    for(int i = 1; i <= n; i++) cin >> a[i];
    for(int i = 1; i <= n; i++) {
        cin >> b[i];
        qe.push(make_pair(a[1]+b[i], i));
        t[i] = 1; //b序列中第i个数和a序列中第1个数相加
    }
    for(int i = 1; i <= n; i ++) {
        printf("%d ", qe.top().first);
        int m = qe.top().second;
        t[m] ++;
        qe.pop();
        qe.push(make_pair(a[t[m]] + b[m], m));
    }
    return 0;
}

学习：
优先队列存放封装好的数据结构pair，比较方式：先按pair的第一个数比较，第一个数相同按第二个数比较

“进阶算法”专栏（目录）