【luogu2602】【bzoj1833】 [ZJOI2010]数字计数 [数位dp]

P2602 [ZJOI2010]数字计数

bzoj1833

套用的写windy数时的数位dp模板 emmm其实还有很多种做法

f[i][j][k]表示填了i位数其最高位数字为j时数码k出现了多少次

注意在 和x的位数相同有一位比x的对应位数小的全部方案数 其余位数对应数字都相同 的时候要再统计一下前面的数用了多少次

有i位数字的情况下 如果不考虑前导0，对于每一个数它们的数量都是相等的，也就是f[i]=f[i-1]*10+10^(i-1)

然后其他大致就是数位dp模板一样了

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<cmath>
#include<stack>
#include<algorithm>
using namespace std;
#define ll long long
#define rg register
const int N=2000000000+5,pw=11,inf=0x3f3f3f3f,P=19650827;
ll a,b;
ll base[20],f[20][10][10];
ll ans[10][2];
template <class t>void rd(t &x)
{
    x=0;int w=0;char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    x=w?-x:x;
}

void pre(){
    base[0]=1ll;
    for(int i=1;i<=13;++i) base[i]=base[i-1]*10ll;
    for(int i=0;i<=9;++i) f[1][i][i]=1ll;
    for(int i=2;i<=13;++i)//填了i个 
    for(int j=0;j<=9;++j)//最高位 
    {
        for(int k=0;k<=9;++k){
            for(int l=0;l<=9;++l)
            f[i][j][l]+=f[i-1][k][l];}
        f[i][j][j]+=base[i-1];
    }
}

void solve(ll x,int y){
    ll p=0,pre=x,num[20];
//    for(p=0;p<=13;++p) if(base[p]>x) break;
    while(x) num[++p]=x%10,x/=10;
    if(!p) return;
    for(int i=1;i<p;++i)
    for(int j=1;j<=9;++j)
    for(int k=0;k<=9;++k) ans[k][y]+=f[i][j][k];
    for(int i=1;i<num[p];++i)
    for(int j=0;j<=9;++j) ans[j][y]+=f[p][i][j];
    for(int i=p-1;i>0;--i){
        for(int j=0;j<num[i];++j)
            for(int k=0;k<=9;++k) ans[k][y]+=f[i][j][k];
        for(int j=p;j>i;--j)
        ans[num[j]][y]+=base[i-1]*num[i];
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("nocows.out","w",stdout);
    rd(a),rd(b);
    pre();
    solve(a,0),solve(b+1,1);
    for(int i=0;i<=9;++i) printf("%lld ",ans[i][1]-ans[i][0]);
    return 0;
}