2442. Sequence （POJ）

题意理解

给定m×n矩阵，一次从每行各取一个数，加起来计算总和，一共有n×m个和，求前n小和的序列。m<500,n<2000

问题分析

用堆数据结构+贪心算法

直接计算m×n个和再排序显得笨。参考网上的思路，对矩阵每行从小到大排序，当m=1时，前n个和等于第一行元素；设m=k时，前k行的前n小和已知。那么当m=k+1时，前k+1的前n小和是，前n小和与k+1行元素逐个相加得到。从左到右依次相加，可以得到，如果某个和超过了前n小和的最大值，那么后面的情况就不用处理了。这里涉及到一个重复操作，每次从前n小和中找到最大那个和，然后替换它。这是用堆的地方。

时间优化

性能优化点1：一次性读入m×n然后逐行拷贝效率低，换成，依次读入一行，原地排序比较高

性能优化点2：遍历前n小和和k+1行的和时，对于超过前n小和最大值出现时就退出遍历。减少无效比较

其他

此题参考网上的，学习了。堆的一个妙用，是一次可以给出一个最大值或是最小值。

用到<algorithm>中的make_heap,pop_heap,push_heap,sort_heap。

对于需要使用break跳出循环的场景，可以将跳出判断放到for循环第二层来实现。

堆是个好东西，有机会再练几题。

代码链接

https://github.com/xierensong/learngit/blob/master/poj/p2442.cpp