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

题目链接<http://acm.hdu.edu.cn/showproblem.php?pid=4507>

---

题意：

求出区间内，满足数位不包含7，数位之和不是7的倍数，本身不是7的步数的所有数字的平方和。

---

题解：

条件是很好满足的，复杂的是求平方和，我们先不求平方和，先考虑求所有数字的一次之和的解法。

对于任何一个数，我们把最高位拆出来，比如123拆成100+23，5764拆成5000+764，这样每个数字都可以拆成x+y的形式。此时x是我们当前枚举的数，y是后面未枚举的数。那么x的贡献就是y的次数。

推广到二次，(x+y)^2=x^2+2*x*y+y^2，x^2的贡献是y的一次和。

所以做三次dfs，分别求出数字的零次和，一次和，二次和就行了。

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=5e5+7;
const ll mod=1e9+7;
ll tt[20];
ll dp[20][10][10][5];
ll dg[20];
ll dfs0(ll p,ll sum,ll m,bool o,bool u){

    if(p==0) return sum&&m&&o;
    if(!u&&~dp[p][sum][m][0]) return dp[p][sum][m][0];
    ll up=u?dg[p]:9,res=0;
    for(ll i=0;i<=up;i++){
        if(i==7) continue;
        res+=dfs0(p-1,(sum+i)%7,(m*10+i)%7,(i==0)?o:1,(u&&(i==up))?1:0);
        res%=mod;
    }
    if(u==0) dp[p][sum][m][0]=res;
    return res;
}
ll dfs1(ll p,ll sum,ll m,ll o,bool u){
    if(p==0) return 0;
    if(!u&&~dp[p][sum][m][1]) return dp[p][sum][m][1];
    int up=u?dg[p]:9,res=0;
    for(ll i=0;i<=up;i++){
        if(i==7) continue;
        ll x=i*tt[p]%mod;
        res+=x*dfs0(p-1,(sum+i)%7,(m*10+i)%7,i==0?o:1,(u&&(i==up))?1:0)%mod;
        res%=mod;
        res+=dfs1(p-1,(sum+i)%7,(m*10+i)%7,i==0?o:1,(u&&(i==up))?1:0);
        res%=mod;
    }
    if(u==0) dp[p][sum][m][1]=res;
    return res;
}
ll dfs2(ll p,ll sum,ll m,ll o,bool u){
    if(p==0) return 0;
    if(!u&&~dp[p][sum][m][2]) return dp[p][sum][m][2];
    int up=u?dg[p]:9,res=0;
    for(ll i=0;i<=up;i++){
        if(i==7) continue;
        ll x=i*tt[p]%mod;
        ll xx=x*x%mod;
        res+=xx*dfs0(p-1,(sum+i)%7,(m*10+i)%7,i==0?o:1,(u&&(i==up))?1:0)%mod;
        res%=mod;
        res+=2*x%mod*dfs1(p-1,(sum+i)%7,(m*10+i)%7,i==0?o:1,(u&&(i==up))?1:0)%mod;
        res%=mod;
        res+=dfs2(p-1,(sum+i)%7,(m*10+i)%7,i==0?o:1,(u&&(i==up))?1:0);
        res%=mod;
    }
    if(u==0) dp[p][sum][m][2]=res;
    return res;
}
ll solve(ll x){
    ll cnt=0;
    while(x){
        dg[++cnt]=x%10;
        x/=10;
    }
    return dfs2(cnt,0,0,0,1);
}
int main(){
    ll t,n,m;
    tt[1]=1; for(ll i=2;i<20;i++) tt[i]=tt[i-1]*10;
    memset(dp,-1,sizeof(dp));
    scanf("%lld",&t);
    while(t--){
        scanf("%lld%lld",&n,&m);
        if(n>m) swap(n,m);
        printf("%lld\n",(solve(m)-solve(n-1)+mod)%mod);
    }
}