PAT A1080(测试点4运行超时)

测试点1，2不过，代码如下：
测试点1，2不过是因为，当前学生的排名要和该学校被录取的最后一个学生的排名一样，当前这个学生才能破格录取进该学校。

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 40010;
struct Stu{
	int id, ge, gi, total, rankPos;
	int pre[6];
}stu[maxn];

vector<int> school;
vector<int> admit[100];

bool cmp(Stu& a, Stu& b) {
	if(a.total != b.total) return a.total > b.total;
	else if(a.ge != b.ge) return a.ge > b.ge;
    return a.id < b.id;
}

int main() {
	int n, m, k;
	cin >> n >> m >> k;
	for(int i=0;i<m;i++) {
		int temp;
		cin >> temp;
		school.push_back(temp);
	}
	for(int i=0; i<n; i++) {
		stu[i].id = i;
		cin >> stu[i].ge >> stu[i].gi;
		stu[i].total = stu[i].ge + stu[i].gi;
		for(int j=0; j<k; j++) {
			cin >> stu[i].pre[j];
		}
	}
	sort(stu, stu+n, cmp);
	stu[0].rankPos = 1;
	for(int i=1;i<n;i++) {                            
		stu[i].rankPos = i+1;
		if(stu[i].ge == stu[i-1].ge && stu[i].total == stu[i-1].total) {
			stu[i].rankPos = stu[i-1].rankPos;
		}
	}
	for(int i=0; i<n; i++) {
		for(int j=0; j<k; j++) {
			if(school[stu[i].pre[j]] > 0) {
				admit[stu[i].pre[j]].push_back(stu[i].id);
				school[stu[i].pre[j]] --;
				break;
			} 
            else {
				int lastId, admitSize;
				admitSize = admit[stu[i].pre[j]].size();
				if(admitSize) {
					lastId = admit[stu[i].pre[j]][admitSize-1];
					if(stu[i].rankPos == stu[lastId].rankPos) {
						admit[stu[i].pre[j]].push_back(stu[i].id);
						break;
					}
				}
			}
		}
	}
	for(int i=0; i<m; i++) {
		if(admit[i].size()) {
			sort(admit[i].begin(), admit[i].end());
			for(int j=0; j < admit[i].size(); j++) {
				if(j) printf(" ");
				printf("%d", admit[i][j]);
			}
		}
		printf("\n");
	}
	return 0;
}

改进代码：(测试点1，2过了 测试点四超时了)

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 40010;
struct Stu{
	int id, ge, gi, total, rankPos;
	int pre[6];
}stu[maxn];
vector<int> school;
vector<int> admit[100];
bool cmp(Stu& a, Stu& b) {
	if(a.total != b.total) return a.total > b.total;
	else if(a.ge != b.ge) return a.ge > b.ge;
	else return a.id < b.id;
}
int main() {
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k); 
	for(int i=0;i<m;i++) {
		int temp;
		scanf("%d", &temp);
		school.push_back(temp);
	}
	for(int i=0; i<n; i++) {
		stu[i].id = i;
		cin >> stu[i].ge >> stu[i].gi;
		stu[i].total = stu[i].ge + stu[i].gi;
		for(int j=0; j<k; j++) {
			scanf("%d", &stu[i].pre[j]);
		}
	}
	sort(stu, stu+n, cmp);
	stu[0].rankPos = 1;
	for(int i=1;i<n;i++) {                            
		stu[i].rankPos = i+1;
		if(stu[i].ge == stu[i-1].ge && stu[i].total == stu[i-1].total) {
			stu[i].rankPos = stu[i-1].rankPos;
		}
	}
	for(int i=0; i<n; i++) {
		for(int j=0; j<k; j++) {
			if(school[stu[i].pre[j]] > 0) {
				admit[stu[i].pre[j]].push_back(stu[i].id);
				school[stu[i].pre[j]] --;
				break;
			} else {
				int lastId;
                int admitSize = admit[stu[i].pre[j]].size();
				if(admitSize) {
					lastId = admit[stu[i].pre[j]][admitSize-1];
                    int index;
                    for(int i=0;i<n;i++) {
                        if(stu[i].id == lastId) {
                            index = i;
                            break;
                        }
                    }
					if(stu[i].rankPos == stu[index].rankPos) {
						admit[stu[i].pre[j]].push_back(stu[i].id);
						break;
					}
				}
			}
		}
	}
	for(int i=0; i<m; i++) {
		if(admit[i].size()) {
			sort(admit[i].begin(), admit[i].end());
			for(int j=0; j < admit[i].size(); j++) {
				if(j) printf(" ");
				printf("%d", admit[i][j]);
			}
		}
		printf("\n");
	}
	return 0;
}

AC代码：(相比较上一个代码，微小改动)
测试点四超时原因：复杂度过高，解决技巧(利用数组小标降低复杂度，即利用数组小标去掉没必要的数组访问)值得熟记，与A 1076“运行超时”的解决方法如出一辙

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 40010;
struct Stu{
	int id, ge, gi, total, rankPos;
	int pre[6];
}stu[maxn];
vector<int> school;
vector<int> admit[100];
bool cmp(Stu& a, Stu& b) {
	if(a.total != b.total) return a.total > b.total;
	else return a.ge > b.ge;
}

bool cmpId(int a, int b) {  // ---- 添加比较函数
    return stu[a].id < stu[b].id;
}

int main() {
	int n, m, k;
	scanf("%d%d%d", &n, &m, &k); 
	for(int i=0;i<m;i++) {
		int temp;
		scanf("%d", &temp);
		school.push_back(temp);
	}
	for(int i=0; i<n; i++) {
		stu[i].id = i;
		cin >> stu[i].ge >> stu[i].gi;
		stu[i].total = stu[i].ge + stu[i].gi;
		for(int j=0; j<k; j++) {
			scanf("%d", &stu[i].pre[j]);
		}
	}
	sort(stu, stu+n, cmp);
	stu[0].rankPos = 1;
	for(int i=1;i<n;i++) {                            
		stu[i].rankPos = i+1;
		if(stu[i].ge == stu[i-1].ge && stu[i].total == stu[i-1].total) {
			stu[i].rankPos = stu[i-1].rankPos;
		}
	}
	for(int i=0; i<n; i++) {
		for(int j=0; j<k; j++) {
			if(school[stu[i].pre[j]] > 0) {
				admit[stu[i].pre[j]].push_back(i);   // 仅仅是改为 保存当前学生的数组小标
				school[stu[i].pre[j]] --;
				break;
			} else {
				int last;
                int admitSize = admit[stu[i].pre[j]].size();
				if(admitSize) {
					last = admit[stu[i].pre[j]][admitSize-1];
					if(stu[i].rankPos == stu[last].rankPos) {
						admit[stu[i].pre[j]].push_back(i); // -----
						break;
					}
				}
			}
		}
	}
	for(int i=0; i<m; i++) {
		if(admit[i].size()) {
			sort(admit[i].begin(), admit[i].end(), cmpId);
            int len = admit[i].size(); 
			for(int j=0; j < len; j++) {
				if(j) printf(" ");
				printf("%d", stu[admit[i][j]].id);  // ------
			}
		}
		printf("\n");
	}
	return 0;
}