洛谷P4127 同类分布数位dp

最新推荐文章于 2020-05-06 19:30:03 发布

SigmaQuadrant

最新推荐文章于 2020-05-06 19:30:03 发布

阅读量128

点赞数

分类专栏：比赛题解

本文链接：https://blog.csdn.net/qq_40859782/article/details/100559462

版权

比赛题解专栏收录该内容

218 篇文章 0 订阅

订阅专栏

给出一个区间 $1\leq l,r \leq 1e18$ ，然后求问有多少数是满足各数位之和是整除原数的。直接记录搜到每一位的数位和 $s u m$ 和数字 $v a l$ ，最后判断 $val\%sum==0$ ，但是 $v a l$ 太大，高达 $1 e 18$ 。注意到数位和的范围最大为 $18 \times 9$ ，所以不妨直接枚举数位和。然后对于固定的数位和，我们就可以把 $v a l$ 在过程中取模。最终只要判断 $s u m$ 是否为给定的 $m o d$ ，并且判断 $v a l$ 是否为 $0$ 就可以了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
int mod=0;
ll dp[20][200][200][2];
int a[20]; 
ll dfs(int pos,int sum,int val,bool lim) {
	if(pos==0) {
		if(sum==mod&&val==0) return 1;
		else return 0;
	}
	if(dp[pos][sum][val][lim]!=-1)
		return dp[pos][sum][val][lim];
	int up=lim?a[pos]:9;
	ll ans=0;
	for(int i=0;i<=up;i++) {
		ans+=dfs(pos-1,sum+i,(val*10+i)%mod,lim&&i==a[pos]);
	}
	return dp[pos][sum][val][lim]=ans;
}

ll solve(ll x) {
	if(x==0) return 0; 
	int len=0;
	while(x) {
		a[++len]=x%10;
		x/=10;
	} 
	ll ans=0;
	for(mod=1;mod<=9*len;mod++) {
		memset(dp,-1,sizeof(dp));
		ans+=dfs(len,0,0,1);
	}
	return ans;
}
int main() {
	ll a,b;
	scanf("%lld%lld",&a,&b);
	printf("%lld\n",solve(b)-solve(a-1));
	return 0;
}