digit-counts

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

您在真实的面试中是否遇到过这个题？  

Yes

样例

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

暴力解法：

class Solution {
public:
    int digitCounts(int k, int n) {
        int ans=0;
        for(int i=0;i<=n;i++){
            ans+=count(i,k);
        }
        return ans;
    }
    int count(int num,int k){
        string tmp=to_string(num);
        int res=0;
        for(int i=0;i<tmp.size();i++){
            if(tmp[i]-'0'==k) res++;
        }
        return res;
    }
};

线性时间复杂度O（n）：

class Solution {
public:
    /*
     * param k : As description.
     * param n : As description.
     * return: How many k's between 0 and n.
     */
    int digitCounts(int k, int n) {
        // write your code here
        if(n==0){
            return k==0?1:0;
        }
        
        int digits=0;
        int p=1;
        while(n>=p){
            digits++;
            p*=10;
        }
        int back=0;
        int cur=n%10;
        int pre=n/10;
        
        int res=0;
        int cnt=1;
        while(cnt<=digits){
            if(cur>k){
                res+=(pre+1)*pow(10,cnt-1);
            }else if(cur==k){
                res+=pre*pow(10,cnt-1)+back+1;
            }else{
                res+=pre*pow(10,cnt-1);
            }
            back=cur*pow(10,cnt-1)+back;
            cur=pre%10;
            pre=pre/10;
            cnt++;
        }
        return k==0?res-10:res;
    }
};