P1631 序列合并

来源:互联网 发布:淘宝教育入口 编辑:程序博客网 时间:2024/06/05 18:27

https://www.luogu.org/problem/show?pid=1631#sub

这里写图片描述

做法:
将 a 和 b 都从小到大排一遍序。
然后组成这样一个矩阵:
a1+b1 , a1+b2 , a1+b3 , …… , a1+bn
a2+b1 , a2+b2 , a2+b3 , ……,a2+bn
a3+b1……
……
……
an+b1 , an+b2 , an+b3 , ……,an+bn

正确性:我们先把每行的头扔进堆里,每行的头是每行中的最小值,所以这样我们能在堆里取到一个最小值。当我们取出第x行的元素时,我们就把它后面那个元素扔进堆里,因为任意时刻每行的头是这一行中最小的,所以这样我们总是能够在堆里找到当前未输出过的最小的值,正确性就能够保证了。
做法是不是很机智呢,但是不是我自己想的啊

#include<iostream>#include<cstring>#include<string>#include<cstdio>#include<cmath>#include<algorithm>#include<queue>using namespace std;int n,a[100009],b[100009];struct H{    int k1,k2;    int x;    bool operator < (const H & a) const{        return a.x<x;    }   };priority_queue <H> q;int main(){    scanf("%d",&n);    for(int i=1;i<=n;i++) scanf("%d",&a[i]);    for(int i=1;i<=n;i++) scanf("%d",&b[i]);    sort(a+1,a+n+1);sort(b+1,b+n+1);    for(int i=1;i<=n;i++)    {        H l;        l.k1=i,l.k2=1,l.x=a[i]+b[1];        q.push(l);    }     int num=0;    while(num<n)    {        H l=q.top();        q.pop();        printf("%d ",l.x);        H p;        p.k1=l.k1;p.k2=l.k2+1;        p.x=a[p.k1]+b[p.k2];        q.push(p);        num++;    }    return 0;}