poj2442

第一次写堆,估计也可以用优先队列做,给一个m*n矩阵,每行任取一个数出来,求最小n 数和。从第一行到第i行,用一个堆去存储前i行,最小的n个i数和。

#include <iostream>
#include <cstring>
#include <string>
#include <queue>
#include <cstdio>
#include <algorithm>
using namespace std;
int a[2005],b[2005],sum[2005];
int main()
{
    int t,m,n;
    scanf("%d",&t);
    while(t--)
    {
        cin>>m>>n;
        for(int i=0;i<n;i++)scanf("%d",&a[i]);
        sort(a,a+n);
       for(int i=1;i<m;i++)
        {
        for(int i=0;i<n;i++)
        scanf("%d",&b[i]);
        sort(b,b+n);
        for(int i=0;i<n;i++)sum[i]=a[0]+b[i];
        make_heap(sum,sum+n);
            for(int j=1;j<n;j++)
                for(int k=0;k<n;k++)
            {
                int temp=a[j]+b[k];
                if(temp>sum[0])break;
                pop_heap(sum,sum+n);
                sum[n-1]=temp;
                push_heap(sum,sum+n);
            }
            for(int i=0;i<n;i++)
                a[i]=sum[i];
        }
        sort(a,a+n);
        for(int i=0;i<n-1;i++)
            printf("%d ",a[i]);
        printf("%d\n",a[n-1]);
    }
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值