hdu 1280 堆排序

来源:互联网 发布:淘宝宝贝重复铺货 编辑:程序博客网 时间:2024/06/05 21:49

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 18816 Accepted Submission(s): 6416

Problem Description
还记得Gardon给小希布置的那个作业么?(上次比赛的1005)其实小希已经找回了原来的那张数表,现在她想确认一下她的答案是否正确,但是整个的答案是很庞大的表,小希只想让你把答案中最大的M个数告诉她就可以了。
给定一个包含N(N<=3000)个正整数的序列,每个数不超过5000,对它们两两相加得到的N*(N-1)/2个和,求出其中前M大的数(M<=1000)并按从大到小的顺序排列。

Input
输入可能包含多组数据,其中每组数据包括两行:
第一行两个数N和M,
第二行N个数,表示该序列。

Output
对于输入的每组数据,输出M个数,表示结果。输出应当按照从大到小的顺序排列。

Sample Input
4 4
1 2 3 4
4 5
5 3 6 4

Sample Output
7 6 5 5
11 10 9 9 8

在看堆排序 写了个裸的堆排序

TLE 代码

#include <iostream>#include <cstdio>#include <algorithm>using namespace std;int a[3005],b[3005*1508];int n,m,t;void down(int i) {    int x,flag=0;    while(i*2<=t&&flag==0) {        if(b[i*2]>b[i])            x=i*2;        else x=i;        if(i*2+1<=t) {            if(b[x]<b[i*2+1])                x=i*2+1;        }        if(x!=i) {            swap(b[x],b[i]);            i=x;        }        else flag=1;    }}void creat() {    for(int i=t/2;i>=1;i--)        down(i);}void heapsort() {    while(t>1) {        swap(b[1],b[t]);        t--;        down(1);    }}int main() {    while(scanf("%d%d",&n,&m)!=EOF) {        t=0;        for(int i=1;i<=n;i++)            scanf("%d",&a[i]);        for(int i=1;i<=n-1;i++)            for(int j=i+1;j<=n;j++) {                b[++t]=a[i]+a[j];            }        creat();        heapsort();        for(int i=1;i<=m;i++) {            printf("%d",b[i]);            if(i!=m) printf(" ");        }        printf("\n");    }    return 0;}

然后看着 找找模板 优化 优化 优化….

AC代码

#include <iostream>#include <cstdio>#include <algorithm>using namespace std;int a[3005],b[3005*1508];int n,m,t;void up(int i) {    int k=i/2;    if(i==1) return;    if(b[k]<b[i]) {        swap(b[i],b[k]);        up(k);    }    return;}void down(int i) {    int x,flag=0;    while(i*2<=t&&flag==0) {        if(b[i*2]>b[i])            x=i*2;        else x=i;        if(i*2+1<=t) {            if(b[x]<b[i*2+1])                x=i*2+1;        }        if(x!=i) {            swap(b[x],b[i]);            i=x;        }        else flag=1;    }}int pop() {    int x=b[1];    swap(b[1],b[t--]);    down(1);    return x;}void ins(int i) {    b[++t]=i;    up(t);}int main() {    while(scanf("%d%d",&n,&m)!=EOF) {        t=0;        for(int i=1;i<=n;i++)            scanf("%d",&a[i]);        for(int i=1;i<=n-1;i++)            for(int j=i+1;j<=n;j++) {                ins(a[j]+a[i]);            }        for(int i=1;i<=m;i++) {            printf("%d",pop());            if(i!=m) printf(" ");        }        printf("\n");    }    return 0;}
原创粉丝点击