基准时间限制: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;
}