1484. 最佳排名 Java题解 （二分，map，排序）【PAT甲级1012】

 

输入样例：

5 6
310101 98 85 88
310102 70 95 88
310103 82 87 94
310104 91 91 91
310105 85 90 90
310101
310102
310103
310104
310105
999999

输出样例：

1 C
1 M
1 E
1 A
3 A
N/A

---

算法思路：

数据存储：

• 对学生信息的存储：用map。其中key存储学生的姓名，value用list存放该学生的四课成绩。
• 对每科成绩的存储，用二维数组，二维数组的每行存储每科的所有成绩。

查找方式：当对每课成绩都排序后，二分查找每个学生的每科成绩在该科成绩数组中的位置。找到名次后，取最小名次并作记录。

Java代码：

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] split = br.readLine().split(" ");
		int n = Integer.parseInt(split[0]);
		int M = Integer.parseInt(split[1]);
		
		HashMap<String, ArrayList<Integer>> map = new HashMap<>();  // 存放每个学生的姓名（字符串），四门成绩（list）
		ArrayList<ArrayList<Integer>> list = new ArrayList<>();  // list.get(i):存放全体学生的某门成绩的
		for(int i = 0; i < 4; i++)  // 初始化list
			list.add(new ArrayList<>());
		
		for(int i = 0; i < n; i++) {
			int []arr = new int[4];
			split = br.readLine().split(" ");
			String name = split[0];
			for(int j = 1; j < 4; j++)
				arr[j] = Integer.parseInt(split[j]);
			
			ArrayList<Integer> l = new ArrayList<>();
			for(int j = 1; j < 4; j++)  arr[0] += arr[j];  // arr[0]中先存放总分，再取平均值
			
			arr[0] = (int)Math.round(arr[0] / 3.0);
					
			for(int j = 0; j < 4; j++) {  // 将成绩分别加入对应的集合
				l.add(arr[j]);  // 每人的成绩集合
				list.get(j).add(arr[j]); //每门成绩的总集合
			}
			map.put(name, l);
		}
		
		for(int i = 0; i < 4; i++) Collections.sort(list.get(i)); // 对每门成绩排序
		
		StringBuilder ans = new StringBuilder();
		for(int i = 0; i < M; i++) {
			String name = br.readLine();
			ArrayList<Integer> value = map.get(name); 
			if(value == null) {  // 不存在该学生
				ans.append("N/A\n");
				continue;
			}
			
			int min = n + 1;
			char c = ' ';
			char []str = new char[] {'A', 'C', 'M', 'E'};
			for(int j = 0; j < 4; j++) {  // 遍历该学生的每门成绩
				int idx = search(list.get(j), value.get(j)); // 找该成绩的对应的排名
				if(idx < min) { // 取最小排名
					min = idx;
					c = str[j];
				}
			}
			ans.append(min + " " + c + "\n");
		}
		System.out.print(ans);
	}
	
	public static int search(ArrayList<Integer> list, int x) { // 二分查找某个数在某个集合中的排名
		int l = 0, r = list.size() - 1;
		while(l < r) {
			int mid = l + r + 1 >> 1;
			if(list.get(mid) <= x) l = mid;
			else r = mid - 1;
		}
		return list.size() - l;  // 由于是从小到大排序的集合，真正的排名需要转换一下
	}
}