题目链接:
题目大意:给出一个m*n的矩阵,从每一行中取出一个数相加。能得到n^m个不同的结果。要求输出当中前n项。
建立一个以n元数组为底层数组的堆,在这里,利用stl中的make_heap,pop_heap。push_heap等函数解决。
1.将第一组数据输入arr1数组。升序排序。
2.将接下来的数据输入到arr2数组中。而且heap[i]=arr1[0]+arr2[0...n-1]。make_heap(heap,heap+n).
3.arr1数组从1到n-1,比較temp=arr1[i]+arr2[0...n-1]与堆顶的元素。假设temp比較小,则将堆顶元素pop。加入temp到heap;否则跳出循环。
4.将heap中的元素所有赋值给arr1数组,升序排序。反复2,3两步,直到所有数据所有处理完。
代码:
#include#include #include #define M 111#define N 2111using namespace std;int arr1[N],arr2[N],heap[N];int m,n;int main(){ int t; bool s=true; scanf("%d",&t); while(t--) { scanf("%d%d",&m,&n); for(int i=0;i