码迷,mamicode.com
首页 > 其他好文 > 详细

最小堆的维护,POJ(2051)

时间:2016-05-11 21:37:16      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:

题目链接:http://poj.org/problem?id=2051

技术分享

技术分享
///维持最小堆(优先队列)POJ2051
#include <iostream>
#include <string>

using namespace std;

struct Node {
    int Now;    ///出堆的时间
    int id;
    int p;  ///时间间隔
};

Node node [3001];
int K;  ///输出个数

void down (Node H[],int s,int m) {
    ///向下调整,s是要调整的编号,m是堆的size
    Node rc=H[s];
    for(int j=s*2; j<=m; j*=2) {
        if(j<m) {
            if(H[j].Now>H[j+1].Now) {   ///两个子节点中找到那个较小者
                j++;
            } else {
                ///如果相等,比较ID
                if(H[j].Now==H[j+1].Now&&(H[j].id>H[j+1].id))
                    j++;
            }
        }
        if(rc.Now<H[j].Now||(rc.Now==H[j].Now&&rc.id<H[j].id))
            break;
        H[s]=H[j];  ///更新子节点
        s=j;
    }
    H[s]=rc;    ///最后交换的子节点换上rc
}

///建立一个最小堆
void MakeMinHeap(Node H[],int length) {
    for(int i=length/2; i>0; --i)
        down(H,i,length);
}

int main() {
    string str;
    int i=1;
    cin>>str;
    while(str.compare("#")!=0) {
        cin>>node[i].id>>node[i].p;
        node[i].Now=node[i].p;
        i++;
        cin>>str;
    }
    int len=i-1;
    cin>>K;
    MakeMinHeap(node,len);
    for(int i=1; i<=K; i++) {
        cout<<node[1].id<<endl;
        node[1].Now+=node[1].p;
        down(node,1,len);   ///修改顶点后再次调整最小堆
    }
    return 0;
}
View Code

 

最小堆的维护,POJ(2051)

标签:

原文地址:http://www.cnblogs.com/TreeDream/p/5483412.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!