【JAVA】PAT 乙级 1055 集体照

每排人数为 N/K（向下取整），多出来的人全部站在最后一排；

后排所有人的个子都不比前排任何人矮；

每排中最高者站中间（中间位置为 m/2+1，其中 m 为该排人数，除法向下取整）；

每排其他人以中间人为轴，按身高非增序，先右后左交替入队站在中间人的两侧（例如5人身高为190、188、186、175、170，则队形为175、188、190、186、170。这里假设你面对拍照者，所以你的左边是中间人的右边）；

若多人身高相同，则按名字的字典序升序排列。这里保证无重名。

现给定一组拍照人，请编写程序输出他们的队形。

输入格式：
每个输入包含 1 个测试用例。每个测试用例第 1 行给出两个正整数 N（≤10⁴​​ ，总人数）和 K（≤10，总排数）。随后 N 行，每行给出一个人的名字（不包含空格、长度不超过 8 个英文字母）和身高（[30, 300] 区间内的整数）。

输出格式：
输出拍照的队形。即K排人名，其间以空格分隔，行末不得有多余空格。注意：假设你面对拍照者，后排的人输出在上方，前排输出在下方。

输入样例：

10 3
Tom 188
Mike 170
Eva 168
Tim 160
Joe 190
Ann 168
Bob 175
Nick 186
Amy 160
John 159

输出样例：

Bob Tom Joe Nick
Ann Mike Eva
Tim Amy John

用一个类来存储人的名字和身高，调用Comparable接口，重写compareTo方法。身高相同按字典序排列可以用字符串的compareTo方法。

	static class person implements Comparable<person> {
		String name;
		int h;

		person(String a, int b) {
			name = a;
			h = b;
		}

		@Override
		public int compareTo(person o) {
			if (this.h == o.h) {
				return this.name.compareTo(o.name);
			}
			return o.h - this.h;
		}
	}

一开始想用字符串储存结果，结果后三个测试点过不了，显示内存超限

	Arrays.sort(p);
	int c = N / K;
	int m = N % K;
	String r = p[0].name;
	for (int i = 1, index = 1; i < c + m; i++, index ^= 1) {
		if (index == 1) {
			r = p[i].name + " " + r;
		} else {
			r = r + " " + p[i].name;
		}
	}
	out.println(r);
	out.flush();
	for (int i = c + m; i < N; i++) {
		r = p[i].name;
		int j = i + 1;
		for (int index = 1; j < i + c; j++, index ^= 1) {
			if (index == 1) {
				r = p[j].name + " " + r;
			} else {
				r = r + " " + p[j].name;
			}
		}
		i = j - 1;
		out.println(r);
		out.flush();
	}

猜测可能是字符串占用了太大内存，所以改进了一下

	Arrays.sort(p);
	int c = N / K;
	int m = N % K;
	for (int i = 0; i < K; i++) {
		int L, R, len;
		person[] tem;
		int index;
		if (i == 0) {
			len = c + m;
			L = len / 2;
			R = len / 2;
			index = 0;
		} else {
			len = c;
			index = c * i + m;
			R = len / 2;
			L = len / 2;
		}
		tem = new person[len];
		tem[R] = p[index];
		R++;
		L--;
		index++;
		while (L >= 0 && R < len) {
			tem[L--] = p[index++];
			tem[R++] = p[index++];
		}
		if (L >= 0) {
			tem[L] = p[index++];
		} else if (R < len) {
			tem[R] = p[index++];
		}
		for (int j = 0; j < len; j++) {
			if (j == 0) {
				out.print(tem[j].name);
			} else {
				out.print(" " + tem[j].name);
			}
		}
		out.println();
	}
	out.flush();

其实也可以在之前的代码上加System.gc()，来释放内存，不过这道题不是很管用，虽然可以多通过几个样例，但是还是有些剩一两个没有过，所以这种取巧的方法一般还是别用吧。

完整AC代码：

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StreamTokenizer;
import java.util.Arrays;

public class Main{
	public static void main(String[] args) throws Exception {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
		StreamTokenizer in = new StreamTokenizer(bf);
		in.nextToken();
		int N = (int) in.nval;
		in.nextToken();
		int K = (int) in.nval;
		person[] p = new person[N];
		for (int i = 0; i < N; i++) {
			in.nextToken();
			String name = in.sval;
			in.nextToken();
			int h = (int) in.nval;
			p[i] = new person(name, h);
		}
		Arrays.sort(p);
		int c = N / K;
		int m = N % K;
		for (int i = 0; i < K; i++) {
			int L, R, len;
			person[] tem;
			int index;
			if (i == 0) {
				len = c + m;
				L = len / 2;
				R = len / 2;
				index = 0;
			} else {
				len = c;
				index = c * i + m;
				R = len / 2;
				L = len / 2;
			}
			tem = new person[len];
			tem[R] = p[index];
			R++;
			L--;
			index++;
			while (L >= 0 && R < len) {
				tem[L--] = p[index++];
				tem[R++] = p[index++];
			}
			if (L >= 0) {
				tem[L] = p[index++];
			} else if (R < len) {
				tem[R] = p[index++];
			}
			for (int j = 0; j < len; j++) {
				if (j == 0) {
					out.print(tem[j].name);
				} else {
					out.print(" " + tem[j].name);
				}
			}
			out.println();
		}
		out.flush();
	}

	static class person implements Comparable<person> {
		String name;
		int h;

		person(String a, int b) {
			name = a;
			h = b;
		}

		@Override
		public int compareTo(person o) {
			if (this.h == o.h) {
				return this.name.compareTo(o.name);
			}
			return o.h - this.h;
		}
	}
}