找到比自己强的人_师徒关系
题目描述:
给定数组[[2,1],[3 2]],
每组表示师徒关系,第一个元素是第二个元素的老师,数字代表排名,
现在找出比自己强的徒弟。
输入输出描述:
输入描述:
输入一个数组
输出描述:
每个元素比自己强的徒弟数量
示例1:
输入:
[[2,1],[3,2]]
输出:
[0,1,2]
说明:
第一行数据[2,1]表示排名第 2 的员工是排名第 1 员工的导师,后面的数据以此类推。
第一个元素 0 表示成绩排名第一的导师,没有徒弟考试超过他;
第二个元素 1 表示成绩排名第二的导师,有 1 个徒弟成绩超过他
第三个元素 2 表示成绩排名第二的导师,有 2 个徒弟成绩超过他
示例2:
输入:
[[2,1],[3,2]]
输出:
[0,1,2]
解题思路:
对于每一个元素而言,比起强的徒弟可能是直系徒弟、也可能是徒弟的徒弟、徒弟的徒弟的徒弟…
故若 A 是 B 的徒弟,且 B 比 A 强,则比 B 强的徒弟 也都强于 A。所以只需要记录下每个元素、比该元素强的徒弟的数量即可(因此使用一个Map来记录每个元素和比他强的数量,key为元素id,value为比他强的数量)。
如下如:示例1的师徒关系表:
排名第1的员工有 0 个比其强的徒弟
排名第2的员工有 1 个比其强的徒弟
排名第3的员工有 2 个比其强的徒弟、徒孙
代码:
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
String[] split = scanner.nextLine().replace("],[", "#").replace("[", "").replace("]", "").split("#");
// 统计每个老师门下有多少个学生超过他
Map<Integer, Integer> map = new TreeMap<>();
for (String s : split) {
String[] item = s.split(",");
int master = Integer.parseInt(item[0]);
int student = Integer.parseInt(item[1]);
// 把学生和老师都记录一下
if (!map.containsKey(master)) {
map.put(master, 0);
}
if (!map.containsKey(student)) {
map.put(student, 0);
}
// 学生比老师强
if (master > student) {
// 学生的学生
Integer allStudent = map.get(student);
map.put(master, map.get(master) + 1 + allStudent);
}
}
// 结果输出
int index = 0;
System.out.print("[");
for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
if (index++ > 0) {
System.out.print("," + entry.getValue());
} else {
System.out.print(entry.getValue());
}
}
System.out.print("]");
}