HUAWEI 机试题：磁盘容量排序

🌈 Scala 实现

题目描述：

磁盘的容量单位有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<l<30)，表示磁盘的容量 有一个或多个格式为mv的子串组成，其中m表示容量大小v表示容量单位 磁盘容量m的范围 1~1024正整数容量单位v的范围包含题目中提到的M,G,T

输出描述:

输出N行 表示N块磁盘容量排序后的结果

示例 输入： 3 1G 2G 1024M 输出： 1G 1024M 2G 说明： 1G和1024M容量相等,稳定排序要求保留他们原来的相对位置，故1G在1024M前 输入： 3 2G4M 3M2G 1T 输出： 3M2G 2G4M 1T 说明： 1T大于2G4M大于3M2G

代码：

def main(args: Array[String]): Unit = {
          
   

    // 获取输入
    val N = StdIn.readLine().toInt
    // 获取每一次磁盘容量
    var buffer = new ListBuffer[String]
    for (i <- 1 to N) {
          
   
        buffer += StdIn.readLine()
    }

    buffer.sortWith((o1, o2) => {
          
   
        val o1F = formatted(o1)
        val o2F = formatted(o2)
        if (o1F != o2F) o1F.compareTo(o2F) < 0 else o1F.compareTo(o2F) > 0
    }).foreach(println)

}

// 实现单位换算，同一换算为 M
def formatted(str: String): Int = {
          
   
    var sum = 0
    var tmpIndex = -1
    for (i <- 0 to str.length - 1) {
          
   
        if (str(i) == M) {
          
   
            sum += Integer.parseInt(str.substring(tmpIndex + 1, i))
            tmpIndex = i
        } else if (str(i) == G) {
          
   
            sum += Integer.parseInt(str.substring(tmpIndex + 1, i)) * 1024
            tmpIndex = i
        } else if (str(i) == T) {
          
   
            sum += Integer.parseInt(str.substring(tmpIndex + 1, i)) * 1024 * 1024
            tmpIndex = i
        }
    }
    sum
}