《leetCode》： Number of Digit One

题目

Given an integer n, count the total number of digit 1 appearing in all non-negative integers less than or equal to n.

For example:
Given n = 13,
Return 6, because digit 1 occurred in the following numbers: 1, 10, 11, 12, 13.

Hint:Beware of overflow.

思路

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){        if(n<0){            return 0;        }        int res = 0;        for(int m=1;m<=n;m*=10){            int a = n/m;            int b = n%m;            int ones = ((a+8)/10)*m;            if(a%10==1){                ones +=(b+1);            }            res += ones;        }        return res;    }

以上的代码还存在一点点的溢出的问题,就如题目中给予的提示一样：

修正的方法为：将int类型的变量变为long类型的变量

修正后的代码如下：

    public class Solution {        public int countDigitOne(int n){            if(n<0){                return 0;            }            int res = 0;            for(long m=1;m<=n;m*=10){                long a = n/m;                long b = n%m;                long ones = ((a+8)/10)*m;                if(a%10==1){                    ones +=(b+1);                }                res += ones;            }            return res;        }    }