问题引入:如何把k个有序的表合并成一个有序的表(假设每个表都是升序排列)
解法:用优先队列来维护每个表的当前元素,如果一共有n个元素,则时间复杂度为O(n*logk)
题目:uva11997
题意
给你k个集合,每个集合里有k个元素,在每个数组里取一个元素加起来,求这些和里最小的k个值
重要代码:
class num
{
public:
int res;
int count;
num(int x,int y)
{
this->res=x;
this->count=y;
}
bool operator<(const num& x) const
{
return x.res<this->res;
}
};
int n;
void merge(int *a,int *b,int *c)
{
priority_queue<num>q;
for(int i=0;i<n;i++)
{
num n(a[i]+b[0],0);
q.push(n);
}
for(int i=0;i<n;i++)
{
num m=q.top();
c[i]=m.res;
q.pop();
if(m.count+1<n)
{
num k(m.res-b[m.count]+b[m.count+1],m.count+1);
q.push(k);
}
}
}