LintCode 3. 统计数字(Java实现，拒绝暴力破解)

描述

计算数字 k 在 0 到 n 中的出现的次数，k 可能是 0~9 的一个值

样例

样例 1：

输入：
k = 1, n = 1
输出：
1
解释：
在 [0, 1] 中，我们发现 1 出现了 1 次 (1)。
样例 2：

输入：
k = 1, n = 12
输出：
5
解释：
在 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] 中，
我们发现 1 出现了 5 次 (1, 10, 11, 12)(注意11中有两个1)。

代码

public int digitCounts(int k, int n) {
        // write your code here
        int result = 0;
        int base = 1; //位, 个位/十位/百位
        if (k == 0 && n == 0) {
            return 1;
        } else if (n < 10 && k < n) {
            /**
             * 说明，lintcode对于在5(n)里面找3(k)的个数这种情况，默认返回1不是0，否则无法AC
             */
            return 1;
        }

        /**
         * 统计方法就是对于 100 里面找 1 的个数
         * 先找个位为1的，1、11、21、31......91
         * 然后找十位为1的，11、12、13、14......19
         * 最后找百位为1的，100
         * 
         * 以此类推
         */
        while (n / base > 0) {
            int cur = (n / base) % 10;
            int low = n - (n / base) * base;
            int hight = n / (base * 10);

            if (k == 0 && cur > k) {
                if (hight != 0) {
                    result += hight + 1;
                } else {
                    result += hight;
                }
            } else if (cur == k) {
                result += hight * base + low + 1;
            } else if (cur < k) {
                result += hight * base;
            } else {
                result += (hight + 1) * base;
            }

            base *= 10;

        }
        return result;
    }