[BZOJ 1799] self 同类分布

Link:

BZOJ 1799 传送门

Solution:

一句话的题目，看得爽，做得烦

一般这类和数位相关的都是数位$dp$吧

不过一开始还是感觉不太可做，毕竟每个数模数不同

但要发现，模数最高也只可能为$9*19=171$，

于是只要将数按照他们的数位和（即模数）分类计算即可

这样便暴力解决了模数不同的问题

设$dp[sp][sum][rmd][lmt]$表示：

枚举到第$sp$高位，剩下的数的和位$sum$，此时对$mod$余$rmd$时的方案数（lmt表示是否达到上界）

感觉数位$dp$还是用记忆化搜索写起来逻辑比较清晰吧

Code:

#include <bits/stdc++.h>
 
using namespace std;
typedef long long ll;
const int MAXN=; //n可能为19，sum最大为171
int vis[][MAXN][MAXN][],mod,dgt,cur,num[];
ll a,b,dp[][MAXN][MAXN][];
 
ll dfs(int sp,int sum,int rmd,int lmt)
{
    if(!sp) return !sum && !rmd;
    if(vis[sp][sum][rmd][lmt]==cur) return dp[sp][sum][rmd][lmt];
    vis[sp][sum][rmd][lmt]=cur;ll ret=;
 
    int l=max(,sum-(sp-)*),r=min((lmt)?num[sp]:,sum);
    for(int i=l;i<=r;i++)
        ret+=dfs(sp-,sum-i,(rmd*+i)%mod,lmt&(i==num[sp]));
    return dp[sp][sum][rmd][lmt]=ret;
}
 
ll solve(ll x)
{
    ll ret=;
    for(dgt=;x;x/=) num[++dgt]=x%;
    for(mod=;mod<=dgt*;mod++)
        cur++,ret+=dfs(dgt,mod,,);
    return ret;
}
 
int main()
{
    scanf("%lld%lld",&a,&b);
    printf("%lld\n",solve(b)-solve(a-));
    return ;
}

Review:

1、少用$memset$，尽量在使用时顺便初始化

为了区分不同次的调用，可以在每一次调用打上不同的标记

2、在$dp$时难以处理模数不同的情况

考虑将数据分类后直接暴力所有可能的模数

3、$1e18$可能有19位，$MAXN$要设为175

（一开始扫了一眼题解上$MAXN$为165，以后还是要自己算啊……）