codevs 1245 最小的n个和

题目:     这里

思路:  考虑n^2中不同情况的样子:

          a1 + b1 , a1 + b2 , a1 + b3 , a1 + b4......a1+bn

          a2 + b1 , a2 + b2 , a2 + b3 , a2 + b4.......a2+bn

          a3 + b1 , a3 + b2 , a3 + b3 , a3 + b4......a3+bn

          a4 + b1 , a4 + b2 , a4 + b3 , a4 + b4.......a4+bn

          a5 + b1 , a5 + b2 , a5 + b3 , a5 + b4.......a5+bn

注意右下的永远大于左上的，首先把第一行扔进堆里面，然后每次取出一个数字(ai+bj)，就把它的下面的一个位置的数字(ai+1,bj)扔进堆里面，这样可以保证每次取出来的都是当前的最小值，因为对于ai+bj,如果下一个数字是aibk(k > i)，那么ai-1+bk一定在这之前出过队列。也就是说下一个数字一定是在堆里面的,所以这样重复n次就可以了.

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 101001;
struct Node{
    int first,second,val;
    bool operator <(const Node x) const{
        return this->val > x.val;
    }
    Node(int first , int second , int val){
        this->first = first; this->second=second;this->val=val;
    }
};

int a[maxn] , b[maxn] , n , ans[maxn];
int main()
{
   while(~scanf("%d",&n)){
    priority_queue<Node>pq;
    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 + 1 + n);
    sort(b + 1, b + 1 + n);
    for(int i = 1; i <= n;i++){
        Node node =  Node(1,i,a[1] + b[i]);
        pq.push(node);
    }
    int tot = 0;
     for(int i = 1; i <= n;i++){
        Node node = pq.top();
        pq.pop();
        ans[tot++] = node.val;
        node = Node(node.first+1,node.second,a[node.first + 1]+b[node.second]);
        pq.push(node);
     }
     for(int i = 0 ; i < tot;i++){
        printf("%d",ans[i]);
        if(i == tot - 1) printf("
");
        else printf(" ");
     }
   }
}