A+B Problem

A+B Problem

终于调试通过了 zhx 讲的数位 DP 的引例:求0~n之间有多少个数(虽然感觉有点傻)

不过仔细一想还是有用处的!可以用来计算\(a+b\)!

即使测试数据中有负数也没关系

//Corona's A+B Problem
//数位DP
//2018-10-07
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int MAXN=18;
long long f[MAXN][2];
inline long long dp(long long n)
{
    int a[MAXN],len=0;
    memset(f,0,sizeof(f));
    memset(a,0,sizeof(a));
    while(n){a[++len]=n%10;n/=10;}
    for(int i=1,j=len;i<j;i++,j--) swap(a[i],a[j]);
    f[0][1]=1;//超出最高位的只能填0,此时与上限的关系为相等 : 都是0
    for(int i=1;i<=len;i++)//从高位开始第 i 位数字
        for(int j=0;j<=1;j++)//枚举当前状态是从f[i-1][0] 还是f[i-1][1]转移而来
            for(int c=0;c<=9;c++)//第i 位填 c.
            if(j==1){
                if(c>a[i]) continue;
                else f[i][c==a[i]] += f[i-1][1];
            }
            else f[i][0] += f[i-1][0]; 
    return f[len][0]+f[len][1];
}

inline int scanf()
{
    int f=1,x=0;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return f*x;
}

inline void printf(int x)
{
    if(x<0){putchar('-');x=~(x-1);}
    int s[13]={0},&top=s[0];
    while(x){s[++top]=x%10;x/=10;}
    if(!top)s[++top]=0;
    while(top)putchar(s[top--]+'0');
}
int main()
{
    int a=scanf(),b=scanf(),x=0,y=0;
    x=dp(abs(a))-1;
    y=dp(abs(b))-1;
    if(a<0) x=(~x)+1;
    if(b<0) y=(~y)+1;
    printf(x+y);
    return 0;
}