P1631 序列合并 ( 堆

按照我原本的思路，是维护一个堆，其中元素为{i，j , val}  同时用set进行对{i，j}进行去重

对于一个（i，j） 如果他是第k个小的数，那么（i+1，j） 和（i,j+1）就有可能是第k+1小的数字

于是将这俩个数字加入堆中（当然要去重啦

#include <bits/stdc++.h>
using namespace std;
using PII = pair<int,int >;
using ll  = long long;
using namespace std;
int n,m;

struct node{
    int i,j,val;
    bool operator < (const node & u)const{
        return val == u.val ? i < u.i:val > u.val;
    }
};
int a[100010];
int b[100010];
int t;
int main(){
    //priority_queue<PII,vector<PII>,greater<PII>> pq;
    priority_queue<node> pn;
    set<PII> se;
    vector<int> c;
    cin>>n;
    for(int i=0;i<n;i++)cin>>a[i];
    for(int i=0;i<n;i++)cin>>b[i];
    pn.push({0,0,a[0]+b[0]});
    se.insert({0,0});
    while(c.size()<=n){
        auto x = pn.top();
        pn.pop();
        int pi = x.i;
        int pj = x.j;
        //cout<<x.i<<" "<<x.j<<" ";
        c.push_back(x.val);
        if(se.find({pi,pj+1}) == se.end() && pj+1 < n){
             se.insert({pi,pj+1});
            pn.push({pi,pj+1,a[pi] + b[pj+1]});
        }
        if(se.find({pi+1,pj}) == se.end() && pi+1 < n){
            se.insert({pi+1,pj});
            pn.push({pi+1,pj,a[pi+1] + b[pj]});
        }
        //cout<<"\n";

    }

    for(int i=0;i<n;i++){
        printf("%d ",c[i]);
    }

}

——————————————————————————————————————————

#include <bits/stdc++.h>
using namespace std;
using PII = pair<int,int >;
using ll  = long long;
using namespace std;
int n,m;
int a[100010];
int b[100010];
int t;
int main(){
    priority_queue<PII,vector<PII>,greater<PII>> pq;
    vector<int> c;
    cin>>n;
    for(int i=0;i<n;i++)cin>>a[i];
    for(int i=0;i<n;i++)cin>>b[i];
    for(int i=0;i<n;i++){
        pq.push({a[0] + b[i], 0 });
    }
    while(!pq.empty() && c.size()<=n){
        auto x = pq.top();
        pq.pop();
        int s = x.first;
        int p = x.second;
        c.push_back(s);
        pq.push({s + a[p+1] - a[p], p + 1});
    }
    for(int i=0;i<n;i++){
        printf("%d ",c[i]);
    }

}

从别人那得到的启发，先将a[0] + b[i] 加入堆中 这就想打与，把所有的i ，j+1加入到了堆中，

那么从堆中取出一个数字后，只要将i+1 , j 在放入堆中就就可以了。