在编写一道算法题时用到了优先级队列,其中优先级比较函数第一次写错了,输出结果一直不对,感觉这个错误应该大家都很容易犯,总结一下前车之鉴。
题目大意是有一个触发器,每个一个周期会产生一个数,触发器是通过输入“Register NUM Period”定义的,输入语句可以包含若干行定义register语句外加整数k,然后按时间先后输出触发器产生的k个整数,若同一时间产生多个整数则先输出数小的。
根据题意,用优先级队列可以比较好的完成这个题目,定义一个结构体变量,包含num,period,time三个属性,另外重载运算符<,以判断优先级。在这个函数中要特别注意函数参数的写法以及函数体的写法,很容易出错,这是最为关键的一部分。
代码如下
#include<iostream> #include<queue> using namespace std; struct Item { int qnum,period,time; bool operator < (const Item& a) const{ return time>a.time || (time==a.time&&qnum>a.qnum); } }; int main(){ priority_queue<Item> pq; char s[20]; while(cin>>s&&s[0]!='#') { Item item; cin>>item.qnum>>item.period; item.time=item.period; pq.push(item); } int k; cin>>k; while(k--) { Item r = pq.top(); pq.pop(); cout<<r.qnum<<endl; r.time=r.time+r.period; pq.push(r); } return 0; }
作者:u011613729 发表于2013-9-22 13:09:09 原文链接
阅读:20 评论:0 查看评论