Scala趣味题之求s=a+aa+aaa+aaaa+aa...a的值

题目为:

求s=a+aa+aaa+aaaa+aa...a的值，其中a是一个数字（1-9之间），n  是数字的个数。例如a为2，n为5，则s=2+22+222+2222+22222

 def main(args: Array[String]): Unit = {
    //控制台获取数值
    println("请输入a")
    var a = StdIn.readInt()
    println("请输入n")
    var n = StdIn.readInt()
    //定义总和初始值与a2，
    var sum = 0
    var a2 = a
    //从0开始遍历，取10的i次方
    for (i <- 0 to n-1){
      var num = math.pow(10,i).toInt;
      //如果是第一次进入，那么a2的值不变，还是本身
      if (i!=0){
        //a的值依旧为2 2*10 2*100 2*1000... 再加a2的值
        //
        a2 += a*num
      }
      //求和2+20 2+20+200 2+20+200+2000
      sum+=a2
    }

    print(sum)

  }