1056 Mice and Rice (25分)
题目链接:PAT A 1056
题目大意:输入第一行给出两个数np和ng,分别代表老鼠总数和每组老鼠的个数,第二行给出每只老鼠(索引从0到np-1)的重量,第三行给出初始时老鼠的分组序号。规则是从每组中选出最重的一只老鼠晋级比赛,这样有几组,就会有几只老鼠晋级,其余老鼠的排名就是组数加一,要求输出最后每只老鼠的排名。如果像样例那样,有11只老鼠,但是每组有3只老鼠,那么前九只老鼠是每三只一组,而后面所有老鼠为一组,凑成四组。
思路分析:建立一个结构体数组,里面保存老鼠的重量和排名。由于这道题需要随时淘汰某些老鼠,并且把晋级的老鼠重新排名,很符合队列的特点,可以用STL中的queue去解题。把老鼠索引按照输入第三行的顺序放进队列里,写一个while循环,条件是队列元素个数不为1(因为如果元素个数为1,说明只剩一只老鼠,一定是第一名,不用分组了),然后确定组数。当np%ng等于0时,group就等于np/ng,否则等于np/ng+1(此时最后一组需要加额外的判别条件,具体看代码)。对于每一组老鼠,先令k等于队首元素的索引,之后依次比较组内老鼠重量与k老鼠的重量,如果大于k老鼠的重量,就更新k,比较完一只老鼠就把这只老鼠从队列里踢出去,同时排名赋值为组数加一,当一组老鼠比较完毕后,把k老鼠重新入队,队列结构中,这只重新入队的老鼠将排在末尾,这也是我们使用队列的原因。之后循环重复上述工程,最后输出即可。
AC代码:
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
struct mouse{
int weight, rank;
};
int main() {
int np, ng, index, group;
scanf("%d %d", &np, &ng);
vector<mouse> v(np);
for(int i = 0; i < np; i++)
scanf("%d", &v[i].weight);
queue<int> q;
for(int i = 0; i < np; i++) {
scanf("%d", &index);
q.push(index); //按题目给定顺序入队
}
while(q.size() != 1) {
if(np % ng == 0) //老鼠数正好
group = np / ng;
else //老鼠数不够,需要加判断条件
group = np / ng + 1;
for(int i = 0; i < group; i++) {
int k = q.front(); //队首元素索引
for(int j = 0; j < ng; j++) {
if(i * ng + j == np) //判断条件,判断老鼠数是否超过给定数字,超过则跳出循环
break;
if(v[q.front()].weight > v[k].weight)
k = q.front();
v[q.front()].rank = group + 1; //排名为组数加一
q.pop(); //将这只老鼠移出队列
}
q.push(k); //把重量最大的老鼠重新入队
}
np = group; //更新晋级的老鼠数
}
v[q.front()].rank = 1; //还在队列中的老鼠一定是第一名
for(int i = 0; i < v.size(); i++) {
printf("%d", v[i].rank);
if(i != v.size() - 1)
printf(" ");
}
return 0;
}