優先佇列的多路歸併問題
阿新 • • 發佈:2019-02-10
問題引入:如何把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);
}
}
}