codeup|问题 B: 序列合并

题目描述
有两个长度都为N的序列A和B，在A和B中各取一个数相加可以得到N2个和，求这N2个和中最小的N个。
输入
第一行一个正整数N（1 <= N <= 100000）。
第二行N个整数Ai，满足Ai <= Ai+1且Ai <= 109
第三行N个整数Bi，满足Bi <= Bi+1且Bi <= 109
输出
输出仅有一行，包含N个整数，从小到大输出这N个最小的和，相邻数字之间用空格隔开。
样例输入 Copy
3
2 6 6
1 4 8
样例输出 Copy
3 6 7
提示
建议用最小堆实现。

代码

#include<bits/stdc++.h>

using namespace std;

struct node {
    int i;
    int data;

    friend bool operator<(node a, node b) {
        //优先队列区别于普通队列的是：优先级队列如果插入的结点是结构体类型，咋需要在结构体中重载比较操作符函数
        return a.data > b.data;
    }
};

int a[100010] = {0};
int b[100010] = {0};

int main() {
    int n;
    while (scanf("%d", &n) != EOF) {
        priority_queue<node, vector<node>> q;
        //优先队列的定义<数据类型，保存数据的容器，元素的比较方式>，当不写后两个参数时，默认容器使用vector，比较方式为operator <
        getchar();
        for (int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
        }
        sort(a + 1, a + n + 1);//注意参数，因为数组下标从1开始
        getchar();
        for (int i = 1; i <= n; i++) {
            scanf("%d", &b[i]);
        }
        sort(b + 1, b + n + 1);
        for (int i = 1; i <= n; i++) {
            node temp;
            temp.data = a[i] + b[1];
            temp.i = 1;
            q.push(temp);
        }
        int m = n;
        while (m--) {
            node t = q.top();
            q.pop();
            printf("%d", t.data);
            if (m >= 1) printf(" ");
            else printf("\n");
            t.data = t.data + b[t.i + 1] - b[t.i];
            t.i++;
            q.push(t);
        }
    }
    return 0;
}