【BZOJ1799 /Ahoi2009】self 同类分布

                               1799: [Ahoi2009]self 同类分布

                                                 Time Limit: 50 Sec  Memory Limit: 64 MB
                                                             Submit: 2188  Solved: 977

Description

给出a,b，求出[a,b]中各位数字之和能整除原数的数的个数。

Input

Output

Sample Input

10 19

Sample Output

3

HINT

【约束条件】1 ≤ a ≤ b ≤ 10^18

 

解析：

       数位DP。

       枚举各位数之和，令表示到第 i 位此时各位数之和为 sum，模枚举的和为 p 时的合法数量。

       这样做复杂度照理说是过不去的，然而不加任何优化最裸的DFS都能跑过去，加上一些优化减枝后跑得更快，所以做题还是得有信仰啊！

 

代码：

#include <bits/stdc++.h>
using namespace std;

const int Max=160;
int m,num[Max];
long long l,r,n,f[20][Max][Max];

inline long long dfs(int pos,int sum,int p,int mod,int limit)
{
	if(!pos) return sum==mod&&!p;
	if(sum>mod||sum+pos*9<mod) return 0;    //优化
	if(!limit&&~f[pos][sum][p]) return f[pos][sum][p];  //记忆化
	int mx=limit?num[pos]:9;
	long long ans=0;
	for(int i=0;i<=mx;++i) ans+=dfs(pos-1,sum+i,(p*10+i)%mod,mod,limit&(i==mx));
	if(!limit) return f[pos][sum][p]=ans;
	return ans;
}

inline long long solve(long long x)
{
	if(x<=0) return 0;
	n=m=0;
	while(x)num[++m]=x%10,x/=10;
	for(int i=1;i<=9*m;++i)  //注意从1开始
	{
	  memset(f,-1,sizeof(f));
	  n+=dfs(m,0,0,i,1);
	}
	return n;
}

int main()
{
	cin>>l>>r;
	cout<<solve(r)-solve(l-1)<<"\n";
	return 0;
}