有序表的最小和(小根堆)

题意:

给出两个长度为n的有序表A和B,在A和B中各任取一个元素,可以得到n2个和,求这些和中最小的n个。(不要去重)

数据范围:n<=4e5

解法:

以n=3为例,下表是两两配对的所有情况:
在这里插入图片描述
上表中(i,j)格子中的数一定小于等于(i,j+1),因为只有a[i]是不变的只有b[j]进行变化,而b[]是有序的
因为表中每一行都是有序的,因此第1小的数一定出现在第1列里面:
在这里插入图片描述
假设第1小的数为(1,1),记录一下答案,现在要求第2小的,如何求出第二小的呢?
可以仿照上面的想法,因为当前最小的一定是每行最左边的那一列里面最小的,现在去除了(1,1)
那么第1行最左边就是(1,2)了,如下图,对这三个数取min就是第2小的数:
在这里插入图片描述
不断进行类似操作直到取出n个数即可。
动态取min的操作可以用小根堆实现。
小根堆的每个节点记录{值,横坐标,纵坐标}

总结:
根据表的性质可以发现当前最小一定是表中每行最左边一列的所有数中的最小值
每次取出一个数的时候,就将这个数在表中的右边一个数加入待选队伍
每次取出的最小值就是当前最小值。

code:
#include<bits/stdc++.h>
using namespace std;
const int N=4e5+5;
int a[N],b[N],n;
struct Node{
    int v,i,j;
    friend bool operator<(Node a,Node b){
        return a.v>b.v;
    }
};
signed 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]);
    priority_queue<Node>q;
    for(int i=1;i<=n;i++){
        q.push({a[i]+b[1],i,1});
    }
    for(int i=1;i<=n;i++){
        printf("%d\n",q.top().v);
        int x=q.top().i,y=q.top().j;
        q.pop();
        if(y+1<=n){
            q.push({a[x]+b[y+1],x,y+1});
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值