【LeetCode-233】Number of Digit One

一看到这道题就想到了动态规划，但是tle了，先看看我的python代码吧！

class Solution(object):
    # 最容易想到的方法了（动态规划），但是tle了
    def countDigitOne(self, n):
        """
        :type n: int
        :rtype: int
        """
        if n < 1:
            return 0

        assist = [0 for i in range(n + 1)]

        assist[1] = 1

        for i in range(10,n + 1):
            assist[i] = assist[i // 10] + assist[i % 10]

        return sum(assist)

后来实在不会了，发现了别人的优秀代码

public class Solution {
    /**
     * intuitive: 每10个数, 有一个个位是1, 每100个数, 有10个十位是1, 每1000个数, 有100个百位是1.  做一个循环, 每次计算单个位上1得总个数(个位,十位, 百位).  
     * 例子:以算百位上1为例子:   假设百位上是0, 1, 和 >=2 三种情况: 
     * case 1: n=3141092, a= 31410, b=92. 计算百位上1的个数应该为 3141 *100 次.
     * case 2: n=3141192, a= 31411, b=92. 计算百位上1的个数应该为 3141 *100 + (92+1) 次. 
     * case 3: n=3141592, a= 31415, b=92. 计算百位上1的个数应该为 (3141+1) *100 次.
     * 以上三种情况可以用 一个公式概括:
     * (a + 8) / 10 * m + (a % 10 == 1) * (b + 1);
     * 
     */
    public int countDigitOne(int n) {  
        int ones = 0;  
        for (long m = 1; m <= n; m *= 10) {  
            long a = n / m, b = n % m;  
            ones += (a + 8) / 10 * m;  
            if(a % 10 == 1) 
                ones += b + 1;  
        }  
        return ones;  
    }  
}