hdu4507（恨7不成妻）数位DP

hdu4507（恨7不成妻）数位DP+求平方和

传送门
题目大意:
如果一个整数符合下面3个条件之一，那么我们就说这个整数和7有关——
　　1、整数中某一位是7；
　　2、整数的每一位加起来的和是7的整数倍；
　　3、这个整数是7的整数倍；
询问区间[l,r]不符合条件数字的立方和
思路
如果求询问区间[l,r]不符合条件数字的个数就是普通的数位dp，但此题需要维护立方和。
所以开个结构体
｛cnt,sum,qsum｝dp[pos][sta][sum]
dp[pos][sta][sum]表示长度pos，数位和对7取余为sta，数对7取余为sum的种类数
cnt为该集合里的种类数，sum为该集合种类数的和，qsum为该集合内的平方和
状态转移为
new.sum=last.sum+(i * 10^pos) * last.cnt;
new.qsum=last.qsum+(2 * i * 10 ^ pos * last.sum)+((i * 10^pos )^2*last.cnt);

#pragma GCC optimize(3,"Ofast","inline")  	//G++
#include<bits/stdc++.h>
#define mem(a,x) memset(a,x,sizeof(a))
#define debug(x) cout << #x << ": " << x << endl;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fcout cout<<setprecision(4)<<fixed
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pii;

const ll inf=0x3f3f3f3f;
const ll mod=1e9+7;
const ll maxn = 1e6+5;

template<typename T> void read(T &x){
    x = 0;char ch = getchar();ll f = 1;
    while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
    while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
template<typename T, typename... Args> void read(T &first, Args& ... args) {
    read(first);
    read(args...);
}

struct node{
    ll cnt,sum,qsum;
}dp[20][20][20];
ll M[20],a[20];
void init(){
    M[1]=1;
    for(ll i=0;i<20;i++)
        for(ll j=0;j<20;j++)
            for(ll k=0;k<20;k++)
                dp[i][j][k].cnt=-1,dp[i][j][k].sum=dp[i][j][k].qsum=0;
    for(ll i=2;i<20;i++) M[i]=M[i-1]*10%mod;
}
node dfs(ll pos,ll sta,ll sum,ll limit){
    if(!limit&&dp[pos][sta][sum].cnt!=-1) return dp[pos][sta][sum];
    if(pos==0){
        if(sta==0||sum==0) return {0,0,0};
        return {1,0,0};
    }
    ll up=limit?a[pos]:9;
    node res={0,0,0};
    for(ll i=0;i<=up;i++){
        if(i==7) continue;
        node t=dfs(pos-1,(sta+i)%7,(sum*10+i)%7,limit&&i==a[pos]);
        res.cnt=(res.cnt+t.cnt)%mod;
        res.sum=(res.sum+t.sum+i*M[pos]*t.cnt%mod)%mod;
        //new.sum=last.sum+(i*10^pos)*last.cnt;
        res.qsum=(res.qsum+t.qsum+((M[pos]*M[pos]%mod*i*i)%mod*t.cnt%mod)+(2*i*M[pos]%mod*t.sum%mod))%mod;
        //new.qsum=last.qsum+(2*i*10^pos*last.sum)+((i*10^pos)^2*last.cnt);
    }
    if(!limit) dp[pos][sta][sum]=res;
    return res;
}
ll solve(ll n){
    ll pos=0;
    while(n){
        a[++pos]=n%10;
        n/=10;
    }
    return dfs(pos,0,0,1).qsum;
}

main(){
    init();
    ll T;
    read(T);
    while(T--){
        ll l,r;
        read(l,r);
        cout<<(solve(r)-solve(l-1)+mod)%mod<<"\n";
    }
}