HUAWEI 机试题：字符串的解压缩

🌈 Scala 实现

题目描述：

有一种简易压缩算法：针对全部为小写英文字母组成的字符串， 将其中连续超过两个相同字母的部分压缩为连续个数加该字母 其他部分保持原样不变. 例如字符串aaabbccccd 经过压缩变成字符串 3abb4cd 请您编写解压函数,根据输入的字符串,判断其是否为合法压缩过的字符串。 若输入合法则输出解压缩后的字符串，否则输出字符串"!error"来报告错误。

输入描述：

输入一行，为一个ASCII字符串 长度不超过100字符 用例保证输出的字符串长度也不会超过100字符

输出描述：

若判断输入为合法的经过压缩后的字符串 则输出压缩前的字符串 若输入不合法 则输出字符串"!error"

示例 输入： 4dff 输出： ddddff 说明： 4d扩展为4个d ，故解压后的字符串为ddddff 输入： 2dff 输出： !error 说明： 2个d不需要压缩 故输入不合法 输入： 4d@A 输出： !error 说明： 全部由小写英文字母组成的字符串，压缩后不会出现特殊字符@和大写字母A 故输入不合法

代码：

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

    // 获取输入
    val line = StdIn.readLine().toList

    // 去除异常情况
    for (elem <- line) {
          
   
        if ((elem < a | elem > z) & (elem < 3 | elem > 9)) {
          
   
            println("!error")
            return
        }
    }

    // 记录数字及压缩的字符，以及最终输出
    var tmpNum = 0
    var tmpChar = -
    var res = new StringBuffer()


    for (i <- 0 until line.length) {
          
   

        var elem = line(i)

        // 处理压缩情况
        if (elem >= 3 & elem <= 9) {
          
   

            tmpNum = elem.toInt - 48 // 参考 ASCII 表，字符会转为十进制数
            tmpChar = line(i + 1)

            // 循环加入该字符，实际只加入 n-1 个字符，因为 else 也会加入一个
            for (j <- 0 to tmpNum - 2) {
          
   
                res.append(tmpChar)
            }

            // 还原
            tmpNum = 0
            tmpChar = -

        } else {
          
   

            // 非压缩字符
            res.append(elem)
        }
    }

    // 输出结果
    println(res)
}