您的位置:首页 > 其它

BZOJ 1799 [Ahoi2009]self 同类分布【数位dp

2016-12-01 10:04 501 查看
枚举数字总和为多少(设为mod),f[i][j][k][0..1]表示算到第i位,前i位的和为j,这个数的数值%mod=k,是否顶上界

转移就是f[i+1][j+x][(k*10+x)%mod] += f[i][j][k]

这样的(顶上界的特判一下嘛,上面那个公式就省略0..1那一维了【滑稽

手癌把i打成了x,调了一年2333

#include<bits/stdc++.h>
using namespace std; long long L,R;
long long f[20][200][200][2];
long long Lim = 162;

char s[25];
inline long long dp(long long a, int mod){
if(a<=0) return 0;
memset(f,0,sizeof f);
memset(s,0,sizeof s);
sprintf(s+1,"%lld",a);
f[0][0][0][1] = 1;
int lth = strlen(s+1);
for(register int i=0;i<lth;++i){
for(register int j=0;j<=mod;++j){
for(register int k = 0;k<mod;++k){
if(!f[i][j][k][0] && !f[i][j][k][1]) continue;
if(f[i][j][k][1]){
f[i+1][j+s[i+1]-'0'][(k*10 + s[i+1] - '0')% mod][1] = 1;
for(register int x=0;x<s[i+1] - '0' && j+x<=mod;++x)
++f[i+1][j+x][(k*10 + x)%mod][0];
}
for(register int x = 0;x<=9&&j+x<=mod;++x)
f[i+1][j+x][(k*10 +x)%mod][0] += f[i][j][k][0];
}
}
}

return f[lth][mod][0][0] + f[lth][mod][0][1];
}

long long ans = 0;

int main(){
// freopen("1.in","r",stdin);
scanf("%lld%lld",&L,&R);
if(R==1000000000000000000ll) ++ans , --R;
for(register int i = 1;i<=Lim;++i)
ans += dp(R,i) - dp(L-1,i);

printf("%lld",ans);
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: