【PAT-A1109】Group Photo

【题意】

N个人拍照的时候排阵形，共K排，前面每排N/K人，有多的都放最后一排。后排的人身高不能比前面的任意一个人矮，同时每排中间的人最高，两边身高依次下降。身高相同的按照名字的字典序排。给出每个人的姓名和身高，输出最终排列的名字阵形。

【代码】

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <string>
using namespace std;
const int maxn = 10010;
struct Person {
	string name;
	int height;
}persons[maxn], mat[11][maxn];

bool cmp(Person &a, Person &b) {
	if(a.height != b.height) {
		return a.height > b.height;
	}
	else {
		return a.name < b.name;
	}
}
int main() {
	int n, k;
	scanf("%d%d", &n, &k);
	for(int i = 0; i < n; i++) {
		cin >> persons[i].name >> persons[i].height;
	}
	sort(persons, persons + n, cmp);
	int row = n/k;
	int lastRow;
	if(n%k == 0) {
		lastRow = n/k;		
	}
	else {
		lastRow = n - n / k * (k- 1);
	}
	int mid = lastRow/2 + 1;
	int ptr = 0, cnt = 0;
	for(int i = mid; i > 0 && i <= lastRow;) {
		mat[k][i] = persons[ptr++];
		cnt++; 
		if(cnt%2 == 1) {
			i = mid -(i - mid)- 1;
		}
		else {
			i = mid +(mid - i);
		}
	}
	mid = row/2 + 1;
	for(int i = k - 1; i > 0; i--) {
		cnt = 0;
		for(int j = mid; j > 0 && j <= row;) {
			mat[i][j] = persons[ptr++];
			cnt++; 
			if(cnt%2 == 1) {
				j = mid -(j - mid)- 1;
			}
			else {
				j = mid +(mid - j);
			}
		}
	}
	for(int i = 1; i <= lastRow; i++) {
		if(i != 1) {
			printf(" ");
		}
		printf("%s", mat[k][i].name.c_str());
	}
	printf("\n");
	for(int i = k - 1; i > 0; i--) {
		for(int j = 1; j <= row; j++) {
			if(j != 1) {
				printf(" ");
			}
			printf("%s", mat[i][j].name.c_str());
		}
		printf("\n");
	}
	
	return 0;
}