HDU - 4507 吉哥系列故事——恨7不成妻 数位DP

题意：

       如果一个整数符合下面3个条件之一，那么我们就说这个整数和7有关——
　　1、整数中某一位是7；
　　2、整数的每一位加起来的和是7的整数倍；
　　3、这个整数是7的整数倍；
　　现在问题来了：吉哥想知道在一定区间内和7无关的数字的平方和。

分析：

          dp[len][sum1][sum2] 表示长度为len对7取模为sum1，各位上的数字和为sum2有多少个满足的数
          一个是与7无关的数的个数，就是简单的数位DP了，很常规。
          第二个与7无关的数的和的维护需要用到第一个个数。
          处理到第pos个数位时，加上i*10^pos * 后面的个数
          第三个的维护需要用到前面两个
          (pre*10^pos + next)^2= (pre*10^pos)^2+2*pre*10^pos*next +next^2

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll mod=1e9+7;

struct node{
    ll cnt,sum,sqsum;
}dp[22][11][11];

ll bit[22];
ll p[22];

node dfs(int len,int sum1,int sum2,bool lim){
    if(len<0){
        node tmp;
        tmp.cnt=(sum1&&sum2);
        tmp.sum=tmp.sqsum=0;
        return tmp;
    }
    if(!lim&&dp[len][sum1][sum2].cnt!=-1) return dp[len][sum1][sum2];
    node ans,tmp;
    int end=lim?bit[len]:9;
    ans.cnt=ans.sum=ans.sqsum=0;
    for(int i=0;i<=end;i++){
        if(i==7) continue;
        tmp=dfs(len-1,(sum1+i)%7,(sum2*10+i)%7,lim&&i==end);
        ans.cnt+=tmp.cnt;
        ans.cnt%=mod;
        ans.sum+=(tmp.sum+i*p[len]%mod*tmp.cnt%mod)%mod;
        ans.sum%=mod;
        ans.sqsum+=(tmp.sqsum+2*p[len]%mod*i%mod*tmp.sum%mod)%mod;
        ans.sqsum%=mod;
        ans.sqsum+=(tmp.cnt*p[len]%mod*p[len]%mod*i%mod*i%mod);
        ans.sqsum%=mod;
    }
    if(!lim) dp[len][sum1][sum2]=ans;
    return ans;
}

ll solve(ll n){
    int len=0;
    while(n){
        bit[len++]=n%10;
        n/=10;
    }
    return dfs(len-1,0,0,1).sqsum;
}

int main(){

    for(int i=0;i<22;i++){
        for(int j=0;j<11;j++){
            for(int k=0;k<11;k++){
                dp[i][j][k].cnt=-1;
            }
        }
    }
    p[0]=1ll;
    for(int i=1;i<=20;i++){
        p[i]=(p[i-1]*10)%mod;
    }
    int T;
    scanf("%d",&T);
    while(T--){
        ll l,r;
        scanf("%lld%lld",&l,&r);
        printf("%lld\n",((solve(r)-solve(l-1))%mod+mod)%mod);
    }
    return 0;
}