1. 程式人生 > 實用技巧 >P2602 [ZJOI2010]數字計數

P2602 [ZJOI2010]數字計數

自己獨立做出來的第一道數位dp(雖然一共做了3道qwq

#include <bits/stdc++.h>
#define inf 2333333333333333
#define N 1000010
#define p(a) putchar(a)
#define For(i,a,b) for(long long i=a;i<=b;++i)
//by war
//2020.7.14
using namespace std;
long long l,r,t;
long long f[20][20][20],ans[20];
vector<long long>dim,ki;

struct node{
    
long long k[10]; }g[20][20][20]; void in(long long &x){ long long y=1;char c=getchar();x=0; while(c<'0'||c>'9'){if(c=='-')y=-1;c=getchar();} while(c<='9'&&c>='0'){ x=(x<<1)+(x<<3)+c-'0';c=getchar();} x*=y; } void o(long long x){ if(x<0){p('-');x=-x;}
if(x>9)o(x/10); p(x%10+'0'); } long long dfs(long long x,long long st,long long op){ if(!x) return 1; if(~f[x][st][op]){ For(i,0,9) ans[i]+=g[x][st][op].k[i]; return f[x][st][op]; } long long maxx=op?dim[x]:9,r=0,k=0; For(i,0,9) g[x][st][op].k[i]=ans[i]; For(i,
0,maxx){ if(st==11 && i==0) dfs(x-1,11,op&(i==maxx)); else{ r+=dfs(x-1,i,op&(i==maxx)); ans[i]+=t*r; k+=r; r=0; } } For(i,0,9) g[x][st][op].k[i]=ans[i]-g[x][st][op].k[i]; return f[x][st][op]=k; } void deal(long long x){ memset(f,-1,sizeof f); memset(g,0,sizeof g); dim.clear(); dim.push_back(666); while(x){ dim.push_back(x%10); x/=10; } dfs(dim.size()-1,11,1); } signed main(){ in(l);in(r); t=1; deal(r); t=-1; deal(l-1); For(i,0,9) o(ans[i]),p(' '); return 0; }