digit count

class  {
    
     * param k : As description.
     * param n : As description.
     * return: An integer denote the count of digit k in 1..n
     */
    public int digitCounts(int k, int n) {
        // write your code here
        int count = 0;
        
        for (int i = 0; i <= n; i++){
            count += countK(i,k);
        }
        
        return count;
    }
    
    private int countK(int i, int k){
        if (i == 0 && k == 0)
            return 1;
            
        int count = 0;
        
        while (i > 0){
            count += i % 10 == k ? 1 : 0;
            i /= 10;
        }
        
        return count;
    }
};