剑指offer-整数中1出现的次数（从1到n整数中1出现的次数）

题目描述
求出1 ~ 13的整数中1出现的次数,并算出100~ 1300的整数中1出现的次数？为此他特别数了一下1~13中包含1的数字有1、10、11、12、13因此共出现6次,但是对于后面问题他就没辙了。ACMer希望你们帮帮他,并把问题更加普遍化,可以很快的求出任意非负整数区间中1出现的次数（从1 到 n 中1出现的次数）。
思路
见博客：https://blog.csdn.net/yi_Afly/article/details/52012593

class Solution(object):
    def core(self,n):
        if n < 1 :
            return 0
        r = n
        base = 1
        count = 0
        while r > 0:
            weight = r % 10
            r //= 10
            count += r*base
            if weight == 1:
                count += (n%base)+1
            elif weight > 1:
                count += base
            base *= 10
        return count

测试用例

if __name__=='__main__':
    s = Solution()
    print(s.core(534))