1. 程式人生 > >優先佇列的多路歸併問題

優先佇列的多路歸併問題

問題引入:如何把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);
       }
   }
}