统计数字问题Python

问题描述

统计数字问题：一本书的页码从自然数1开始顺序编码直到自然数n。书的页码按照通常的习惯编排，每个页码都不含多余的前导数字0。例如第6页用6表示而不是06或006。数字统计问题要求对给定书的总页码，计算出书的全部页码中分别用到多少次数字0,1,2,3,…9。
输入：自然数n
输出：0-9每个数字出现的次数
请设计算法，用python编程实现。要求给出分析过程，程序代码，测试数据，测试结果。

思考

思路很简单，先记录第一个$0$的个数，然后计算高位出现次数依次向低位即可。
复杂度为$O(nlogn)$

参考代码

def get_sum(p):
    count = []
    for i in range(10):
        count.append(0)
    for i in range(p + 1):
        temp = i
        if i == 0:
            count[0] += 1
        while temp != 0:
            x = temp % 10
            count[x] += 1
            temp = temp // 10
    for j in range(10):
        print("%d出现%d次" % (j, count[j]))


def main():
    p = eval(input())
    get_sum(p)


if __name__ == "__main__":
    main()