Problem:POJ2442 - Sequence

Problem:POJ2442 - Sequence
Begin Time : 2nd/March/2012 1:00 p.m.

End Time: 2nd/March/2012 4:14 p.m

Cost Time: 3H 14Min
看的别人的解题报告过的，非常感谢 http://hi.baidu.com/%C0%B6%C9%ABarch/blog/item/f9d343f49cd92e53d7887d73.html
的博主！
思路：
我们要找到n个smallest的数，用贪心法可以解决这一问题。

(1)维护两个数组，a和b，以及一个大根堆p
循环不变式：
（1）初始化
将元素读入a，将a排序（从小到大）
执行并重复（2）
（2）保持
对于这全部数据第二行到第m行（从第二行开始，因为第一行读到a里了）
将元素读入b，将b排序（从小到大）
For i = 0 to n -1
heap.push(a[i]+b[0]);
然后
for(int i = 1; i < n;i++)

{
for(int j = 0 ; j < n; j++)

{
if( (b[i] + a[j]) > heap.top() ) break;

heap.pop(); ///从heap中删除一个最大的元素，从而保证heap中元素数目不变

heap.push(b[i] + a[j]);

}
}
/////这样，就选出了n个最小值
然后把heap中的n个值按照从小到大给
a[1] -> a
,并将heap清空。
执行（2）
（3）最终结果
输出a中全部元素就可以了，注意，对于每个case都要换行哦！

 

#include "stdafx.h"

#include <cstdio>  

#include <cstring>  

#include <cstdlib>  

#include <iostream>  

#include <vector>  

#include <queue>

#include<conio.h>

#include <algorithm>  

//#define INPUT  

using namespace std; 

bool comp(int a,int b) 

{ 

    return b > a; 

} 

int main ()

{

vector <int>a;

vector<int>b;

int x;

cin>>x;

while (x--)

{

 int m,n;

   a.clear();

   b.clear(); 

 cin>>m>>n;

 for(int i=0;i<n;i++)

 {

  int buf;

  cin>>buf;

  a.push_back (buf);   

 }

  sort(a.begin(),a.end(),comp); 

    priority_queue <int> q;

 for(int j=0;j<m-1;j++)

 {

        for(int i=0;i<n;i++)

  {

    int buf;

    cin>>buf;

    b.push_back (buf); 

  }

   sort(&b[0],&b[0]+n,comp); 

          for(int z=0;z<n;z++)

    {

             q.push (a[z]+b[0]);

    }

           for(int i=1;i<n;i++)

     {

      for(int z=0;z<n;z++)

      {

       if(q.top()>a[z]+b[i])

       {

                     q.pop();

      q.push(a[z]+b[i]);

       }

       else break;

      }

     }

     for(int z=0;z<n;z++)

     {

      a[n-z-1]=q.top();

      q.pop();

     }

 }

    for( int z=0;z<n-1;z++)

     {

     cout<<a[z]<<" ";

     }

cout<<a[n-1];

 cout<<endl;

 getch();

 

}

 return 0;

}