BZOJ 1799 [Ahoi2009]self 同類分佈【數位dp

枚舉數字總和爲多少（設爲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;
}