TOPK算法

#include<bits/stdc++.h> 

using namespace std;
const int N=10010;
int a[N],b[N],res[N];
struct node{
    int x,y,sum;
    bool operator < (const node &a)const{ //改为小顶堆
        return sum>a.sum;
    }
};
int main( )
{
    priority_queue<node>que;//默认大顶堆
    int n,k;
    scanf("%d%d",&n,&k);
    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++){
        int sum=a[i]+b[1];
        que.push({i,1,sum});
    }
    while(k--){
        int x=que.top().x;
        int y=que.top().y;
        int res=que.top().sum;
        printf("%d ",res);
        que.pop();
        que.push({x,y+1,a[x]+b[y+1]});
    }
    return 0;
}

思路以及相关知识点提炼：

1.小根堆与大根堆

在stl中priority_queue这个数据结构是已经实现好了的堆，且默认是大根堆，即最大值在顶上。

这里先复习下几个常用的接口函数，pop(),push(),top(),size()，差不多也够用，毕竟使用这个数据结构的重点是利用其自动排序功能。

还有就是当数据结构中放的是结构体类型时，要重置一下小于号，当然有很多种实现方法，只要熟练使用一种就行，上面代码中是在结构体中实现的，若还是要实现大根堆，那么就把下方改为小于号。

2.算法优化

在做题时遇到的第一个问题就是怎么把和收集起来，最朴素的方法是两层循环，也就是o（n^2）的时间复杂度，由数据范围可知会TLE。那么就需要优化算法，这里我们能发现其实没有必要维护n^2组数据，因为只要求前k小数，所以其实只要维护前k个小数就可以了。发现将数组排序后，和的值有特定的规律。（TOPK算法——一般都是通过priority_queue来实现）

 变化：将2组数据换为n组数据且数据范围减小，可以直接利用上一题的思路，先整合两个数组，再依次添加进新的数组。

#include<bits/stdc++.h> 

using namespace std;
const int N=205;
int a[N],b[N],c[N];
int m,k;
struct node{
    int x,y,sum;//记录下标和总和
    bool operator < (const node &a)const{
        return sum>a.sum;
    }
};
priority_queue<node>res;
void TopK(){
    int tmp=k;
    int cnt=0;
    for(int i=1;i<=k;i++){
        res.push({i,1,a[i]+b[1]});
    }
    while(tmp--){
        int x=res.top().x;
        int y=res.top().y;
        int sum=res.top().sum;
        //printf("%d ",sum);
        c[++cnt]=sum;
        //printf("%d ",c[cnt]);
        res.pop();
        res.push({x,y+1,a[x]+b[y+1]});
    }
}
int main( )
{
    int n;
    scanf("%d%d%d",&n,&m,&k);
    int tmp=k;
    //先输入两组数据，然后其前k小数构成新数组，再与新输入的数组合并
    for(int i=1;i<=m;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&b[i]);
    }
    sort(a+1,a+m+1);      //预处理前两组数据
    sort(b+1,b+m+1);
    for(int i=1;i<=m;i++){
        res.push({i,1,a[i]+b[1]});
    } 
    int cnt=0;
    while(tmp--){
        int x=res.top().x;
        int y=res.top().y;
        int sum=res.top().sum;
        c[++cnt]=sum;
        res.pop();
        res.push({x,y+1,a[x]+b[y+1]});
    }
    while(n-2){               //依次添加进新的数组
        while(!res.empty()){
            res.pop();
        }
        for(int i=1;i<=m;i++){
            scanf("%d",&a[i]);
            //printf("%d ",a[i]);
        }
        sort(a+1,a+m+1);
        for(int i=1;i<=k;i++){
            b[i]=c[i];
            //printf("%d ",b[i]);
        }
        TopK();
        n--;
    }
    for(int i=1;i<=k;i++){
        printf("%d ",c[i]);
    }

    return 0;
}