HDU 4389 X mod f(x) (数位DP)

题目链接 ：

http://acm.hdu.edu.cn/showproblem.php?pid=4389

题意 ：问[l,  r]之间满足这个数是其各个位置上的数之和的整数倍这一条件有多少个。

思路 ： 数位DP， 因为一个数各个位置上的数之和是 <= 81的， 所以可以枚举最后的模，而这个模也是其所有位置上的数字之和。

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      

using namespace std;
const int maxn = 90;
int bit[20], dp[11][maxn][maxn][maxn];

void init(){
    memset(dp, -1, sizeof(dp));
}

int dfs(int len, int mod, int sum, int res, int fp){
    if (!len){
        return sum == mod && !res;
    }
    if (!fp && dp[len][mod][sum][res] != -1){
        return dp[len][mod][sum][res];
    }
    int Max = (fp ? bit[len] : 9), ret = 0;
    for (int i = 0; i <= Max; i++){
        ret += dfs(len - 1, mod, sum + i, (res * 10 + i) % mod, fp && i == Max);
    }
    if (!fp)dp[len][mod][sum][res] = ret;
    return ret;
}

int solve(int x){
    if (!x) return 0;
    int len = 0;
    while (x){
        bit[++len] = x % 10;
        x /= 10;
    }
    int sum = 0;
    for (int i = 1; i <= 81; i++){
        sum += dfs(len, i, 0, 0, 1);
    }
    return sum;
}

int main(){
    int T; init();
    scanf("%d", &T);
    for (int cas = 1; cas <= T; cas++){
        int l, r;
        scanf("%d%d", &l, &r);
        printf("Case %d: %d\n", cas, solve(r) - solve(l - 1));
    }
    return 0;
}

     
     
    
    
   
   

PS ： 一开始就想到了枚举mod，但是考虑了下一个数有可能是两个mod的倍数的话会重复计算， 似乎要用容斥来搞， 而容斥的复杂度是指数级的， 后来发现自己真的逗比了。

sum == mod不就好了啊