给定n， 求出小于n的所有数中1的位数

看到qiang.xu的博客中得每日一题 http://www.cnblogs.com/xuqiang/archive/2010/11/20/1953393.html，然后也看到给出的算法，正确性无疑，但是算法复杂度随着输入数字的增大了成正比增加，以至于输入是百万千万的时候就很难算出来了，于是重新设计了一下算法，还没有太仔细验证正确性。

原作者给出的代码

#include <stdio.h>
#include <stdlib.h>
// 给定一个i，然后返回i中数字1的个数，比如
// 1251返回的是2
int countTotal1Nums(int i)
{
    int count = 0;
    while(i > 0)
    {
        if ((i % 10) == 1)
        {
            ++count;
        }
        i = i / 10;
    }
    return count;
}
// 给定一个数n，该函数返回的是[1, n]中n个数中
// 所有1的个数
 int count(int n)
 {
     int total = 0;
     int i;
     for(i = 1; i <= n; ++i)
     {
        total += countTotal1Nums(i);
     }
     return total;
 }
int main()
{
    // 下面是测试程序
    // 2
    printf("n = %d, and the result is %d\n", 10, count(10));
    return 0;
}

改进后代码：

#include <stdio.h>
#include <math.h>

int CountOfOneAt(int n, int i)
{
        int base = (int)pow(10.0,i);
        int high = n / base;
        int low = n % (base / 10);
        int current = n % base / (base/10);

        if(current == 0)
        {
                return high * base / 10;
        }
        else if(current == 1)
        {
                return high * base / 10 + low + 1;
        }
        else
        {
                return (high + 1) * base / 10;
        }
}
int CountOfOne(int n)
{
        int total = 0;
        int i = 1;
        while((n/(int)pow(10.0,i-1)) > 0)
        {
                total += CountOfOneAt(n,i);
                i ++;
        }
        return total;
}

int main(int argc,const char *argv[])
{
        if(argc > 1)
        {
                int num = atoi(argv[1]);
                printf("nums below %d contains totally %d ones.\n",num,CountOfOne(num));
        }

        return 0;
}

改进后的代码复杂度只跟输入数字的位数有关，复杂度大大降低。

测试数字 4321532452324354 

shell:~/dev> time ./a 4321532452324354
nums below 783598594 contains totally 729199620 ones.
0.000u 0.000s 0:00.00 0.0%    0+0k 0+8io 0pf+0w

转载于:https://www.cnblogs.com/zhiyzhan/archive/2011/10/21/2219800.html