统计数字

描述

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

样例

例如n = 12, k = 1, 在[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12], 我们发现1出现了5次(1, 10, 11, 12)

题解

要从1到ABCDE中找到k这个数字在某一位上出现了多少次有以下几种情况：

1. 当某一位的数字小于K时，那么该位出现的次数为：更高位数字 X 当前位数
2. 当某一位的数字等于k时，那么该位出现的次数为：更高位数字 X 当前位数 + 低位数字 + 1
3. 当某一位的数字大于k时，那么该位出现的次数为：（更高位数字 + 1） X 当前位数

代码

public class Solution {
    /*
     * @param : An integer
     * @param : An integer
     * @return: An integer denote the count of digit k in 1..n
     */
    public int digitCounts(int k, int n) {
        // write your code here
        if (k == 0 && n < 10) {
            return 1;
        }
        int temp = n, cnt = 0, pow = 1;
        while (temp != 0) {
            int digit = temp % 10;
            if (digit < k) {
                cnt += (temp / 10) * pow;
            } else if (digit == k) {
                cnt += (temp / 10) * pow + (n - temp * pow + 1);
            } else {
                if (!(k == 0 && temp / 10 == 0)) {
                    cnt += (temp / 10 + 1) * pow;
                }
            }
            temp /= 10;
            pow *= 10;
        }
        return cnt;
    }

};