统计数字问题

首先容易想到暴力法，即对每个数字拆解，计数，但是当n很大时很慢.

优化后的算法可以达到O(logn)的复杂度，明显好于暴力，具体参考：点这里

#include <cstdio>
#include <cstring>
#include <cmath>
#include <ctime>
using namespace std;

int ans1[10], ans2[10];

// 解法1: 暴力法 
void solve1(int n){
	for(int i = 1; i <= n; ++i){
		int k = i;
		do{
			++ans1[k%10];
			k/= 10;
		}while(k != 0);
	}
}

//解法2: 分治
int f[10];			//f[n]: n位数(0-9)排列组合所有结果中, 每个数字出现的次数.  n*10^n / 10
int preZero[10];	//preZero[n]: n位数的前导零个数 
void solve2(int n){
	int len = log10(n) + 1;//位数 
	int highest = n / pow(10, len-1);//最高位
	if(len == 1){//是个位数 
		for(int i = 0; i <= n; ++i) ++ans2[i];
		return;
	}
	
	for(int i = 0; i < 10; ++i) ans2[i]+= highest*f[len-1];
	for(int i = 1; i < highest; ++i) ans2[i]+= pow(10, len-1);
	
	//处理最高位 
	int n1 = n - highest*pow(10, len-1);// 除去最高位的剩余部分
	ans2[highest]+= n1 + 1;
	
	//处理中间的0. 例如: 50035 
	int len2 = log10(n1) + 1;
	int midZero = (len-1) - len2;
	ans2[0] += midZero*(n1+1);
	if(n1 > 10) ans2[0] += pow(10, (int)log10(n1));//加上35的前导零
	
	if(n1 > 0) solve2(n1);
}

int main()
{
	int n;
	memset(f, 0, sizeof(f));
	memset(preZero, 0, sizeof(preZero));
	for(int i = 1; i < 10; ++i) f[i] = i * pow(10, i-1);
	for(int i = 2; i < 10; ++i) preZero[i] = preZero[i-1] + pow(10, i-1);
	while(scanf("%d",&n) == 1&&n){
		
		clock_t start, end;
		
		start = clock();
		solve1(n);
		end = clock();
		
		printf("n = %d\n", n);
		printf("solve 1:\n"); 
		for(int i = 0; i < 10; ++i){
			printf("%d ", ans1[i]);
		}puts("");
		printf("O(nlogn) time: %dms\n", end - start);
		
		start = clock();
		solve2(n);
		end = clock();
		
		int len = log10(n) + 1;
		ans2[0] -= preZero[len-1];
		--ans2[0];//从1开始 
		
		printf("solve 2:\n"); 
		for(int i = 0; i < 10; ++i){
			printf("%d ", ans2[i]);
		}puts("");
		printf("O(logn) time: %dms\n\n", end - start);
	}
	return 0;
}

运行效果：