PAT 1025 PAT Ranking

PAT 1025 PAT Ranking

#include <bits/stdc++.h>

using namespace std;

const int N = 3e4 + 10; 
struct student{
	string id;  // 账号 
	int score;  // 分数 
	int loc_id;  // 考点号
	int loc_r;  // 考点排名
	int rank;  // 总排名 
} stu[N];

// 考点内排序 
bool cmp1(student a, student b) {
	// 这里不能写成 if(a.loc_id == b.loc_id) return a.score > b.score; 会出现错误 
	if (a.loc_id != b.loc_id) {
		return a.loc_id < b.loc_id;
	} else {
		return a.score > b.score;
	}
}

// 全部考点一起排序 
bool cmp2(student a, student b) {
	if (a.score != b.score) {
		return a.score > b.score;
	} else {
		return a.id < b.id;
	}
} 

int main() {

	int n;
	cin >> n;
	int num = 0;
	for (int i = 1; i <= n; i++) {
		int k;
		cin >> k;
		for (int j = 1; j <= k; j++) {
			cin >> stu[num].id >> stu[num].score;
			stu[num++].loc_id = i;
		}
	}
		
	// 考点内排序 
	sort(stu, stu + num, cmp1);
	
	int cnt = 1;  // 记录考点内的人数
	stu[0].loc_r = 1;
	// 求考点内的排名 
	for (int i = 1; i < num; i++) {
		if (stu[i].loc_id != stu[i-1].loc_id) {
			cnt = 1;
			stu[i].loc_r = 1; 
		} else {
			cnt++;
			if (stu[i].score == stu[i-1].score) {
				stu[i].loc_r = stu[i-1].loc_r;
			} else {
				stu[i].loc_r = cnt;
			}
		}
	} 
	// 全部考点一起排序
	sort(stu, stu + num, cmp2);
	// 求总排名
	stu[0].rank = 1;
	for (int i = 1; i < num; i++) {
		if (stu[i].score == stu[i-1].score) {
			stu[i].rank = stu[i-1].rank;
		} else stu[i].rank = i + 1;
	} 
	 
	cout << num << endl; 
	for (int i = 0; i < num; i++) {
		cout << stu[i].id << " " << stu[i].rank << " " << stu[i].loc_id << " " << stu[i].loc_r << endl;
	}

	return 0;
}