Description
已知两个正整数a和b,求在a与b之间(包含a和b)的所有整数的十进制表示中1出现的次数。
Input
多组数据(不超过100000组),每组数据2个整数a,b.(1≤a,b≤1000000).
Output
每组数据的答案占一行。
Sample Input
1 10
10 100
2 1
Sample Output
2
20
1
Time Limit: 3 Sec Memory Limit: 128 Mb
分析:
- 题目中,由于要输入多组数据,可以想象,如果我们暴力求解的话,即数据组数为M,每组区间为N,数据位数为T,时间复杂度为O(MNT)。按题目的要求,最坏情况下会达到10^11次运算,肯定是超时的。所以不能暴力做。
- 观察输入条件会发现,我们在计算每一组的时候,有的数字包含1的个数会被重复计算,所以我们可以先将每个数字包含1的个数算出来。又问的是区间,所以我们可以把每个数及其之前的数的所有1的个数保存起来,在需要使用的时候,可以直接用sum[j] - sum[i]就可以直接找出来了。
注意:
- 由于数据规模较大,我们输入采用scanf()
- 由于不确定输入多少行,所以使用
while(scanf("%d %d", &a, &b) != EOF)
{
//循环体
}
//或
while((cin>>a>>b) != 0)
{
//循环体
}
代码:
#include<iostream>
using namespace std;
int sum[1000001];
int compute(int n)
{
int num = 0;
while(n != 0)
{
if(n%10 == 1)
num++;
n /= 10;
}
return num;
}
int main()
{
sum[0] = 0;
for(int i=1; i<=1000000; i++)
sum[i] += sum[i-1] + compute(i); //预处理
int a, b;
// while(scanf("%d %d", &a, &b) != EOF)
while((cin>>a>>b) != 0)
{
if(a>b)
{
int tmp = a;
a = b;
b = tmp;
}
printf("%d\n", sum[b] - sum[a-1]);
}
return 0;
}
总结:
前缀和就像是拿空间来换时间一样,保存每个结果,后面直接索引这个结果就可以了。有点像动态规划保存子问题的解。
前缀和算法可以描述为:给定整数数组,用O(N)时间预处理,用O(1)时间计算区间和的算法