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;
	}
};