算法题：求s=a+aa+aaa+aaaa+aa…a的值

题目：

求s=a+aa+aaa+aaaa+aa…a的值，其中a是一个数字。
而aa…a 的最大长度是n

思路：

我们假设a 就是那个数字(1 ~ 9) ， 最大长度是n

无非就是求下面的公式的值
${\sum }_{i=1}^n{x}_n$ = $x_1+x_2+x_3...+x_n$

其中
$x_n$ = $\underset{n个}{\underbrace{aaa...a}}$

其实再分析下
$x_n$ = $\underset{n个}{\underbrace{aaa...a}}$ = $a\ast 10^{n-1}$ + $a\ast 10^{n-2}$ + … + $a\ast 10^0$

有了上面这个公式， 解题思路就不难了

import math


def get_sum_aaaa(a: int, n: int):
    """
    :param a: the number, 1 to 9
    :param n: the number of elements of the sum list
    :return: the sum of the int list
    """

    sum_v = 0
    for i in range(1, n + 1):  # for range(1,5) means 1, 2 , 3, 4 but 5 is excluded
        sum_v = sum_v + x(a, i)
    return sum_v


def x(a: int, n: int):
    """
    :param a: the number, 1 to 9
    :param n: the length of aaaa...a
    :return: the value of aaaa...a
    """
    x_value = 0
    for i in range(0, n):
        x_value = x_value + a * math.pow(10, i)
    return int(x_value)


if __name__ == '__main__':
    print("Sum of 2 + 22 + 222 + 222..2 n=10 is {}".format(get_sum_aaaa(2, 10)))

用了双重循环， $O(n^2)$的时间复杂度并不好

另1个思路：

$x_n$ = $\underset{n个}{\underbrace{aaa...a}}$ = $a\ast 10^{n-1}$ + $a\ast 10^{n-2}$ + … + $a\ast 10^0$

在上面的数列中中， $X_n$ =$X_{(n-1)}$ * 10 + a
也就是将我们可以从前一项可以方便推导出后一项的值。

java 代码：

public class SumAAA {
    public static void main(String[] args) {
       System.out.println(MessageFormat.format("Sum of 2 + 22 + 222 + 222..2 n=10 is {0}", new SumAAAUitl().getSum(2,10)));
    }

}

class SumAAAUitl{
    /**
     * 
     * @param a  the number, 1 to 9
     * @param n  the number of elements of the sum list
     * @return the sum of the int list
     */
    public Long getSum(int a, int n){
        Long sum = 0L;
        Long x = 0L;

        for (int i=1; i<=n; i++){
            x = x * 10 + a; // X(n) = X(n-1) * 10 + a
            sum += x;
        }
        return sum;
    }
}

时间复杂度降为$O(n)$