poj2442 堆

标签：堆

让你求出序列和最小的前n个序列的序列和。

先介绍下堆的基本操作：

int arr[N];

make_heap(arr+1,arr+N);//对数组arr建堆，堆里第一个元素为最大值

pop_heap(arr+1,arr+N+1);//将第一个元素与最后一个元素交换

push_heap(arr+1,arr+N+1);//录入元素

sort_heap(arr+1,arr+N+1);//排序

思路：用普通方法超时，用堆做 具体如下：

1.将第一序列读入sum中，排序。

2.将数据读入num向量中排序。

    用arr数组存sum[1] + num[i] ( 1<=i<=n)的和

    用make_heap对arr建堆。

    然后for循环比较sum[1] + num[i] (1<=i<=n)与堆arr的顶点,前者大则退出，否则删除

    堆的顶点，插入前者。

 3.将arr的数据拷贝到sum中，并对sum按升序排序

 4.循环2,3步，直到所有数据读入完毕。

 5.打印sum中的数据即为结果。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <stdlib.h>
#include <string.h>
#define N 3000
#define INF 10000000
#define eps 10E-9
#define mem(a)  memset(a,0,sizeof(a))
#define w(a)   while(a)
#define s(a)   scanf("%d",&a)
#define ss(a,b)   scanf("%d%d",&a,&b)
using namespace std;
int arr[N],num[N],sum[N];
int main()
{
    int n,i,j,k,m,t;
    s(t);
    w(t--)
    {
        mem(sum);
        ss(m,n);
        for(i=1; i<=n; i++)
            s(sum[i]);
        for(i=2; i<=m; i++)
        {
            mem(num);
            sort(sum+1,sum+n+1);
            for(j=1; j<=n; j++)
            {
                s(num[j]);
            }
            sort(num+1, num+n+1);
            for(j=1; j<=n; j++)
                arr[j]=sum[1]+num[j];
            make_heap(arr+1,arr+n+1);
            for(j=2; j<=n; j++)
            {
                for(k=1; k<=n; k++)
                {
                    int cnt=sum[j]+num[k];
                    if(cnt>=arr[1])
                        break;
                    pop_heap(arr+1,arr+n+1);
                    arr[n]=cnt;
                    push_heap(arr+1,arr+n+1);
                }
            }
            for(j=1; j<=n; j++)
                sum[j]=arr[j];
        }
        sort(sum+1,sum+n+1);
        for(i=1; i<n; i++)
            printf("%d ",sum[i]);
        printf("%d\n",sum[i]);
    }
    return 0;
}

poj2442 堆

标签：堆

原文地址：http://blog.csdn.net/bigsungod/article/details/45342093