洛谷P4127同类分布

传送


我们要在dfs的板子里记录哪些量呢?当前填的所有数的和sum?当前填的数构成的数值all?

sum可以留下,数值就扔掉叭。数值最大是1e18,要是留下,在g数组里有一维的大小是1e18。也许可以通过特殊方式使chen_zhe把空间上限放到你能存下。手动滑稽

那么为了我们的空间够用,怎么办呢?

上面不记录数值是因为它太大了,那我们能不能让它取模?

当然可以了,不过是对什么取模呢?1e9+7?显然不行,因为l,r<=1e18,产生重复的太多了

大质数?万一你填出来的数是取模数-1那也凉凉。

我们再考虑一下怎么判断sum能整除填出来的数。sum%mo==all%mo?理论上可行,but我们不能确定模数,所以待定

那如果我们枚举填出来的数字之和mo,使其成为模数,则只需判断sum是否等于mo即可。

然后代码就有了

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int inf=214748364;
const ll mod=1000000007;
inline ll read()
{
    char ch=getchar();
    ll x=0;bool f=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<1)+(x<<3)+(ch^48);
        ch=getchar();
    }
    return f?-x:x;
}
ll l,r;
ll g[20][172][172],li[20];
int t,mo;
void make(ll k)
{
    t=0;
    while(k)
    {
        li[++t]=k%10;
        k/=10;
    }
}
ll dfs(int now,int sum,int all,int mo,bool lim)
{
	if(!now) return (!all)&&(sum==mo);
	if(!lim&&g[now][sum][all]!=-1) return g[now][sum][all];
	int up=lim?li[now]:9;
	ll rtn=0;
	for(int i=0;i<=up;i++)
	{
	    rtn+=dfs(now-1,sum+i,(all*10+i)%mo,mo,lim&&(i==up));
	}
	if(!lim) g[now][sum][all]=rtn;
	return rtn;	
}
int main()
{
   ll ans=0;
   l=read();
   r=read();
   make(r);
   for( mo=1;mo<=9*t;mo++)
    memset(g,-1,sizeof(g)),ans+=dfs(t,0,0,mo,1);
   make(l-1);
   ll qwq=0;
   for( mo=1;mo<=9*t;mo++)
	{
		memset(g,-1,sizeof(g));
	    qwq+=dfs(t,0,0,mo,1);	}
   ans-=qwq;
   printf("%lld",ans);
}
原文地址:https://www.cnblogs.com/lcez56jsy/p/11487305.html