洛谷P1392 取数 [堆]

时间:2024-09-14 08:33:14

  题目传送门

取数

题目描述

在一个n行m列的数阵中,你须在每一行取一个数(共n个数),并将它们相加得到一个和。对于给定的数阵,请你输出和前k小的取数方法。

输入输出格式

输入格式:

第一行,三个数n,m,k。

第2~n+1行,每行m个正整数

输出格式:

一行共k个数,代表在每一行取一个数前k小的加和

输入输出样例

输入样例#1: 复制
3 3 2
1 2 3
6 3 5
4 1 2
输出样例#1: 复制
5 6

说明

对于20%的数据,n≤8

对于100%的数据,n≤800,k≤m≤800


  分析:

  WA了无数次,最后发现就是一个取址符号没打。。。。。。真煞笔。。。

  这个蒟蒻真的懒得写了,如果想看思路就参考一下这里吧,蒟蒻就只上代码了。

  Code:

#include<bits/stdc++.h>
using namespace std;
int n,m,k,a[][];
int size,b[],c,cc;
struct Node{
int x,y,num;
}bdg,h[];
inline void Swap(Node &x,Node &y)
{Node temp=x;x=y;y=temp;}
inline void ins(Node x)
{
h[++size]=x;
int ka=size;
while(ka>){
if(h[ka].num<h[ka>>].num){
Swap(h[ka],h[ka>>]);ka>>=;}
else break;}
}
inline void delet()
{
h[]=h[size--];
int ka=,s=ka<<;
while(s<=size){
if(s<size&&h[s+].num<h[s].num)s++;
if(h[s].num<h[ka].num){
Swap(h[s],h[ka]);ka=s;s=ka<<;}
else break;}
}
inline Node get()
{
Node ret=h[];
delet();return ret;
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=;i<=m;i++)
scanf("%d",&a[c][i]);
sort(a[c]+,a[c]+m+);
for(int j=;j<=n;j++){
cc=-c;size=;
for(int i=;i<=m;i++)
scanf("%d",&b[i]);
sort(b+,b+m+);
bdg.x=;
for(int i=;i<=k;i++){
bdg.y=i;bdg.num=a[c][]+b[i];ins(bdg);}
for(int i=;i<=k;i++){
bdg=get();
a[cc][i]=bdg.num;
bdg.x++;bdg.num=a[c][bdg.x]+b[bdg.y];
ins(bdg);}
c=-c;}
for(int i=;i<=k;i++)
printf("%d ",a[c][i]);
return ;
}