HDU 4507 吉哥系列故事――恨7不成妻(经典数位dp，好题)

题目分析

让你求与7无关的数的平方和，第一个条件不含7，那么我们在dfs进行深搜的时候遇到这种情况直接continue掉，关于第二个和第三个条件都很简单，只需要在状态中加2维，并且在状态转移中添加上即可。但是这些都不是重点，本题让你求与7无关的数的平方和，刚开始看到求得这个东西内心是崩溃的，但是你要想到数位dp是按位进行求解的，因此肯定是一位一位去求，并且将深度较大的节点的值传给深度较小的递归求解。但是因为是求得平方和，按位dp的话平方与平方直接并没有什么实质性的关系，因此需要在dp的每个之中增加2个状态，一个是当前状态下不含7的个数，第二是当前不含7的个数的和，当然还是就是平方和了。这样状态就可以转移了,具体看代码。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long
const int maxn = 45;
const LL mod = 1e9+7;
LL bit[maxn], C[maxn];

struct Node{
    LL num;
    LL sum;
    LL qsum;
    Node() {}
    Node(LL a, LL b, LL c):num(a), sum(b), qsum(c) {}
}dp[maxn][8][8];

Node dfs(int pos, int pre1, int pre2, int limit){
    if(pos < 1) return Node(pre1!=0&&pre2!=0, 0, 0);
    if(!limit && dp[pos][pre1][pre2].num != -1) return dp[pos][pre1][pre2];
    int len = limit?bit[pos]:9;
    Node ret = Node(0, 0, 0);
    for(int i = 0; i <= len; i++){
        if(i == 7) continue;
        Node tmp = dfs(pos-1, (pre1+i)%7, (pre2*10+i)%7, limit && i==len);
        ret.num = (ret.num + tmp.num)%mod;  //这一步很好理解

        ret.sum += (((i*C[pos-1])%mod*tmp.num)%mod+tmp.sum)%mod;  //这一步就是除了上一部分的和，还要加上当前位置表示的数乘以上一部分满足条件的数
        ret.sum %= mod;

        ret.qsum += (tmp.qsum + ((2*i*C[pos-1])%mod*tmp.sum)%mod)%mod;  //(a+b)*(a+b) = a*a+2*a*b+b*b;这一步相当于求了a*a+2*a*b
        ret.qsum %= mod;

        ret.qsum += (((i*C[pos-1]*i)%mod*C[pos-1])%mod*tmp.num)%mod; //这一步相当于求了b*b,别忘了乘以tmp.num
        ret.qsum %= mod;
    }
    if(!limit) dp[pos][pre1][pre2] = ret;
    return ret;
}

LL solve(LL n){
    int len = 0;
    while(n){
        bit[++len] = n%10;
        n /= 10;
    }
    Node ans = dfs(len, 0, 0, true);
    return ans.qsum;
}

void init(){
    C[0] = 1;
    for(int i = 1; i < maxn; i++)
        C[i] = (10LL*C[i-1])%mod;
}

int main(){
    int T;
    scanf("%d", &T);
    init();
    memset(dp, -1, sizeof(dp));
    while(T--){
        LL L, R;
        scanf("%I64d %I64d", &L, &R);
        printf("%I64d\n", (solve(R) - solve(L-1) + mod)%mod);  //有可能有负数，仍需要注意
    }
    return 0;
}