这道题用了Arrays.sort,lambda表达式和三元运算,3种运算法则凝结于一行,十分精妙,解析在该行注释
题目描述:
磁盘的容量单位常用的有M,G,T这三个等级,它们之间的换算关系为1T = 1024G,1G = 1024M,现在给定n块磁盘的容量,
请对它们按从小到大的顺序进行稳定排序(稳定排序是重点!),
例如给定5块盘的容量,1T,20M,3G,10G6T,3M12G9M排序后的结果为20M,3G,3M12G9M,1T,10G6T。注意单位可以重复出现,
上述3M12G9M表示的容量即为3M+12G+9M,和12M12G相等。
输入描述:
输入第一行包含一个整数n(2 <= n <= 100),表示磁盘的个数,接下的n行,每行一个字符串(长度大于2,小于30),表示磁盘的容量,
由一个或多个格式为mv的子串组成,其中m表示容量大小,v表示容量单位,例如20M,1T,30G,10G6T,3M12G9M。
磁盘容量m的范围为1到1024的正整数,容量单位v的范围只包含题目中提到的M,G,T三种,换算关系如题目描述。
输出描述:
输出n行,表示n块磁盘容量排序后的结果。
示例1
输入
3
1G
2G
1024M
输出
1G
1024M
2G
说明
2G的容量大于1024M,1G和1024M容量相等,稳定排序要求保留它们原来的相对位置,故1G在1024M之前
代码
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
public class ArraysSort_Stable_Lambda {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
sc.nextLine();
int[][] num = new int[n][2];//用来放置磁盘索引和单位转换成M之后的数值
Map<Integer, String> map = new HashMap<>();//用来放置磁盘索引和磁盘原数值
for (int i = 0; i < n; i++) {
int sum = 0;
int index = -1;
String str = sc.nextLine();
map.put(i + 1, str);
for (int j = 0; j < str.length(); j++) {
if (str.charAt(j) == 'M') {
sum += Integer.parseInt(str.substring(index + 1, j));
index = j;
} else if (str.charAt(j) == 'G') {
sum += Integer.parseInt(str.substring(index + 1, j)) * 1024;
index = j;
} else if (str.charAt(j) == 'T') {
sum += Integer.parseInt(str.substring(index + 1, j)) * 1024 * 1024;
index = j;
}
}
num[i][0] = i + 1;//磁盘索引
num[i][1] = sum;//转换成M之后的数值
}
//对二维数组进行排序
//用了Arrays.sort,lambda表达式和三元运算
//Arrays.sort(arr,sort),每次取arr中两个元素进行对比,会根据sort中的lambda表达式决定升序还是降序排列
//e1和e2代表每次比较的两个元素,如果是e1-e2,那么这俩会升序排列,反之sort是e2-e1则为降序,0为保持相对位置不变
//详细原理请查Arrays.sort,排序的总体过程暂且略过,这里的升序降序和不变都是指e1和e2的相对位置
//lambda表达式(e1, e2) -> (e1[1] == e2[1] ? 0 : (e1[1] - e2[1]))表示参数为e1和e2的情况下,返回值是e1[1] == e2[1] ? 0 : (e1[1] - e2[1])
//三元运算e1[1] == e2[1] ? 0 : (e1[1] - e2[1])的意思是,a?b:c,如果a为true,返回b,否则返回c
//这里a是e1[1] == e2[1],b是(e1[0] - e2[0]),c是(e1[1] - e2[1])
//因为这次的arr是二维数组,e1和e2是每次对比时从arr中取出的两个元素,根据int[][] num = new int[n][2];可知e1和e2是int[2]的数组,
//又根据之前代码,e1[0]表示num[i][0]是磁盘索引,e1[1]表示num[i][1]是转换成单位M之后的数值,e2同理
//此题中要求根据转换成单位M之后的数值升序要e1-e2,所以M值不同时,根据三元运算选(e1[1] - e2[1])
//题目中还要求稳定排序,那么M值相同e1[1] == e2[1]时,(e1[0] - e2[0])表示根据磁盘索引index升序,这样保留了它们原来的相对位置,符合稳定排序的要求
Arrays.sort(num, (e1, e2) -> (e1[1] == e2[1] ? (e1[0] - e2[0]) : (e1[1] - e2[1])));
for (int i = 0; i < n; i++) {
if (i == n - 1) {
//最后一行不加回车符
System.out.print(map.get(num[i][0]));
} else {
//加回车
System.out.println(map.get(num[i][0]));
}
}
}
}