ACM编程 磁盘容量排序【Java】

题目描述：

磁盘的容量单位常用的有 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 块磁盘容量排序后的结果。

public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);
        String n = sc.nextLine();
        String[] strArr = new String[Integer.parseInt(n)];
        for (int i =0;i<Integer.parseInt(n);i++) {
            strArr[i] = sc.nextLine();
        }

        Map<Integer,List<String>> map = new HashMap<>();
        List<Integer> scList = new ArrayList<>();

        for (String str:strArr) {
            Integer sum = getSum(str);
            scList.add(sum);
            List<String> valList = map.get(sum);
            if(null==valList){
                valList = new ArrayList<>();
            }
            valList.add(str);
            map.put(sum,valList);
        }
        scList.sort(null);

        for (Integer i:scList) {
            List<String> strings = map.get(i);
            for (String str:strings) {
                System.out.println(str);
            }
        }
    }

    /**
     * 计算大小
     * @param str
     * @return
     */
    public static Integer getSum(String str){

        int num = 0;
        StringBuilder numStr = new StringBuilder();
        for (char ch:str.toCharArray()) {
            if(ch == 'M'){
                num = num+Integer.parseInt(numStr.toString());
                numStr = new StringBuilder();
            }else if(ch == 'G'){
                num = num+Integer.parseInt(numStr.toString())*1024;
                numStr = new StringBuilder();
            }else if(ch == 'T'){
                num = num+Integer.parseInt(numStr.toString())*1024*1024;
                numStr = new StringBuilder();
            }else{
                numStr.append(ch);
            }
        }
        return num;

    }