给定k个有序表(不妨假设均为升序),求这k个有序表中最小的n个元素,这类问题称为多路归并问题。
对于这类问题,考虑先取出每个有序表中的最小的元素,在取出的这些元素中的最小值即为所有元素的最小值,必然要被选中。根据每个表的有序性,在某个元素的前一个元素未被选中时,该元素必定不会被选中。因此,每次选中了一个元素后,它的后一个元素才会被考虑。
所以在上述第一批取出的k个元素中,先取出最小的一个,再将其后一个元素加入待选取的元素中,然后考虑新的待选取元素的集合,选取其中最小的元素,以此即可选出第二小的元素。
从而可以考虑构建一个优先队列,先将k个有序数组中的最小值均入队,每次取出队列顶端的元素,再将其在表中的后一个元素加入优先队列,依次选取n次即可得到最小的n个元素。
由于每次队列中都只有k个元素,因此时间复杂度为 O ( n log k ) O(n\log{k}) O(nlogk)
例如Argus
题意:输入一些命令,每个命令有一个事件的编号num与周期period,每过一个周期都会执行一次给定编号的事件,在同一时刻优先执行编号小的事件,求前k个执行的事件。
思路:很显然每个命令就是一个以period为公差的等差数列,形成一个有序表,因此利用上述的方法即可。
AC代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
#define rep(i,s,t) for(int i=(s);i<=(t);i++)
#define per(i,s,t) for(int i=(s);i>=(t);i--)
#define ms0(ar) memset(ar,0,sizeof ar)
#define ms1(ar) memset(ar,-1,sizeof ar)
#define Ri(x) scanf("%d",&x)
#define Rii(x,y) scanf("%d%d",&x,&y)
#define Riii(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define De(x) cout<<"#Debug "<<x<<'\n'
#define Dee(x,y) cout<<"#Debug "<<x<<' '<<y<<'\n'
#define pb push_back
struct Node{
int q,p,t