PAT甲级 1056 Mice and Rice (25分)

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;
}