数字1的数量

1009 数字1的数量 

基准时间限制：1 秒 空间限制：131072 KB 分值: 5 难度：1级算法题

 收藏

 关注

给定一个十进制正整数N，写下从1开始，到N的所有正数，计算出其中出现所有1的个数。

例如：n = 12，包含了5个1。1,10,12共包含3个1，11包含2个1，总共5个1。

Input

输入N(1 <= N <= 10^9)

Output

输出包含1的个数

Input示例

12

Output示例

5

刚开始写数位DP，很代码很生硬，凑出来的正确代码。

#include<map>
#include<stack>
#include<queue>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#define maxn 505
#define inf 0x3f3f3f3f3f
#define ll  long long
#define mod 10
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
int a[20];
ll dp[20];
ll x=0;int n;
ll p(int y,bool f){
    ll ans1=1,ans2;
    double z=pow(10.0,y);
    ans1=(ll)z;
    ans2=n%(ans1*10);
    if(f==0)return ans1;
    if(ans2<ans1&&n!=0)ans2=ans1*10;
    //cout<<"y="<<y<<" "<<n<<" "<<ans1<<" "<<ans2<<" "<<(ans2-ans1>=ans1?ans1:(ans2-ans1+1))<<endl;
    return ans2-ans1>=ans1?ans1:(ans2-ans1+1);

}
ll dfs(int pos,bool limit){
    if(pos==-1)return 0;
    if(!limit&&dp[pos]!=-1)return dp[pos];
    int up=limit?a[pos]:9;
    ll tmp=0;
    for(int i=0;i<=up;i++){
       if(i==1)tmp+=p(pos,limit);
        tmp+=dfs(pos-1,limit&&i==a[pos]);
    }
    if(!limit)dp[pos]=tmp;
    return tmp;
}
ll solve(int n){
    int k=0;
    while(n){
        a[k++]=n%10;
        n/=10;
    }
    x=k-1;
    return dfs(k-1,true);
}
int main(){
    while(cin>>n){mem(dp,-1);mem(a,0);
        x=0;
        ll ans=solve(n);cout<<ans<<endl;
    }
}

别人的数位DP

dp[pos][num]记录的是当前长度前面有num个1的情况下1的总数

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn = 20;
ll dp[maxn][maxn], a[maxn];

ll dfs(int pos, int num, int limit)
{
    if(pos == -1) return num;
    if(!limit && dp[pos][num] != -1) return dp[pos][num];
    int up = limit ? a[pos] : 9;
    ll tmp = 0;
    for(int i = 0; i <= up; i++)
        tmp += dfs(pos-1, num+(i==1), limit && i == a[pos]);
    if(!limit) dp[pos][num] = tmp;
    return tmp;
}

ll solve(ll x)
{
    int pos = 0;
    while(x)
    {
        a[pos++] = x%10;
        x /= 10;
    }
    return dfs(pos-1, 0,  1);
}

int main(void)
{
    ll n;
    memset(dp, -1, sizeof(dp));
    while(cin >> n)
        printf("%lld\n", solve(n));
    return 0;
}

也有数学写法;

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int sum[20];
int getlen(int n)
{
	int num=0;
	while(n) {
		n=n/10;
		num++;
	}
	return num;
}
 
int main()
{
	int n,i,j,t1,t2,len,ans;
	cin>>n;
	len=getlen(n);
	sum[1]=n/10;
	if(n%10>=1) sum[1]++;
	for(i=2,t1=10,t2=100;i<=len;i++,t1=t1*10,t2=t2*10) {
		if(n/t1%10>1) {
			sum[i]=n/t2+1;
			sum[i]=sum[i]*t1;
		}
		else if(n/t1%10==1) {
			sum[i]=n/t2;
			sum[i]=sum[i]*t1+n%t1+1;
		}
		else {
			sum[i]=n/t2;
			sum[i]=sum[i]*t1;
		}
	} 
	ans=0;
	for(i=1;i<=len;i++) ans+=sum[i];
	cout<<ans<<endl;
	return 0;
}