lintcode 3. 统计数字

class lintcode_3 {
public:
	/*
	当某一位的数字小于k时，那么该位出现k的次数为：更高位数字*当前位数(10^k)
	当某一位的数字等于k时，那么该位出现k的次数为：更高位数字*当前位数(10^k)+低位数字+1
	当某一位的数字大于k时，那么该位出现k的次数为：(更高位数字+1)*当前位数(10^k)
	*/
	int digitCounts(int k, int n) {
		if (k < 0 || k > 9) return 0;
		int val = 1, num = k == 0 ? 1 : 0;
		for (int i = 0; i < 32; ++i)
		{
			int nTemp = n % (val * 10);
			int nHigh = n / (val * 10);
			int nMid = nTemp / val;
			int nLow = nTemp % val;
			if (nMid == 0 && nHigh == 0) break;
			num += nHigh * val;
			if (nMid > k) {
				if (k != 0) num += val;
			} else if (nMid == k) {
				if (k != 0) num += nLow + 1;
			}
			val *= 10;
		}
		return num;
	}
};