【JavaScript】7-38 数列求和-加强版 (20 分)

题目：

给定某数字A（1 ≤ A ≤ 9）以及非负整数N（0 ≤ N ≤ 100000），求数列之和S=A+AA+AAA+⋯+AA⋯A（N个A）。例如A=1, N=3时，S = 1 + 11 +111 = 123。

输入格式：

输入数字A与非负整数N。

输出格式：

输出其N项数列之和S的值。

输入样例：

1 3

输出样例：

123

JavaScript代码：

const { parse } = require('path')
var readline = require('readline')

const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
})

rl.on('line', function(line) {    
    let tokens = line.split(' ')
    let A = tokens[0]
    let N = tokens[1]

    let ret = sum(A, N)
    console.log(ret)
})

function sum(A, N) {
    let res = []
    let carry = 0

    for (let i = 0; i < N; i++) {
        let s = A * (N - i) + carry
        res.push(s % 10)
        carry = Math.floor(s / 10)
    }

    if (carry != 0 || N == 0) {
        res.push(carry)
    }
    
    res.reverse()
    return res.join('')
}