BZOJ 1799 self 数位dp

bzoj 1799

题意给你一个区间 让你求区间内的数满足各位上的数相加是原数的一个约数的数有多少个

我们不难想到状态转移方程因为最多有162位 那么我们利用加法的余数定理 将他不断的%已知的p

p从1 到 9 * pos,然后判断的时候也就是pos 到了 -1 我们要看余数是不是0并且 被除数是不是我们定的这个p

用数位dp解决 三维dp

/*
bzoj 1799
*/
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <cmath>
#include <stack>
#include <set>
#include <sstream>
#include <queue>
#include <iostream>
using namespace std;
typedef long long ll;
int a[20];
ll dp[20][162][162];
ll dfs(int pos,int p,int old,int sum,bool limit){
    if(pos==-1) return (old==0&&sum==p);
    if(sum>p||(pos+1)*9+sum<p) return 0;
    if(!limit&&dp[pos][old][sum]!=-1) return dp[pos][old][sum];
    int up = limit?a[pos]:9;
    ll ans = 0;
    for(int i = 0;i<=up;++i){
        ans += dfs(pos-1,p,(old*10+i)%p,sum+i,limit&&i==a[pos]);
    }
    if(!limit) dp[pos][old][sum] = ans;
    return ans;
}

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