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

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

 

题目大意：找出区间内不满足整数中某一位是7或整数的每一位加起来的和是7的整数倍或这个整数是7的整数倍的数的平方和

 

题目思路：如果只是单纯的求解这种数的个数，相信都嗤之以鼻，就是一个模板题，但是这题是求平方和，应该怎么办？首先我们可以发现，如果知道前面的数字是若干数的平方和，如1^2+2^2+3^2，在前面加一个新的数，会怎么样？拿1举例，就会变(10+1)^2+(10+2)^2+(10+3)^2，拆开来就是3*10^2+2*10*(1+2+3)+1^2+2^2+3^2，很轻松的发现它是由三部分组成，第一部分是数字个数*这次枚举的数乘以10^位数，来送他到指定位置(比如对1来说，如果他后面还有俩数，他的指定位置就是100，仨数就是1000，以此类推)，然后进行平方，接下来是2*这家伙的实值*后面的数的和，最后一部分就是后面的数的平方和，所以我们只需记录后面的数的和和后面的数的平方和即可解决问题

 

以下是代码：

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(ll i=a;i<=b;i++)
#define per(i,a,b) for(ll i=a;i>=b;i--)
#define ll long long
const ll MAXN = 100+5;
const ll MOD =1e9+7;
ll a[MAXN],b[MAXN];
struct node{
    ll cnt,sum,sqsum;
}dp[MAXN][7][7];
node dfs(ll pos,ll sum,ll val,ll limit){
    if(!pos){
        node temp;
        temp.cnt=temp.sum=temp.sqsum=0;
        if(!sum||!val){
            return temp;
        }
        temp.cnt++;
        return temp;
    }
    if(!limit&&dp[pos][sum][val].cnt!=-1)return dp[pos][sum][val];
    ll up=limit?a[pos]:9;
    node ans;ans.cnt=ans.sum=ans.sqsum=0;
    rep(i,0,up){
        if(i==7)continue;
        node temp=dfs(pos-1,(sum+i)%7,(val*10+i)%7,limit&&i==a[pos]);
        ans.cnt=(ans.cnt+temp.cnt)%MOD;
        ans.sum=(ans.sum+(temp.cnt*i%MOD*b[pos]%MOD+temp.sum)%MOD)%MOD;
        ans.sqsum=((ans.sqsum+temp.cnt*i%MOD*b[pos]%MOD*i%MOD*b[pos]%MOD)%MOD+2*i*b[pos]%MOD*temp.sum%MOD+temp.sqsum)%MOD;
    }
    if(!limit)dp[pos][sum][val]=ans;
    return ans;
}
ll solve(ll x){
    ll pos=0;
    while(x){
        a[++pos]=x%10;
        x/=10;
    }
    return dfs(pos,0,0,1).sqsum;
}
int main(){
    ll t,l,r;
    scanf("%lld",&t);
    b[1]=1;
    rep(i,2,20)b[i]=b[i-1]*10%MOD;
    memset(dp,-1,sizeof(dp));
    while(t--){
        scanf("%lld%lld",&l,&r);
        ll p=solve(r)-solve(l-1);
        cout<<(p%MOD+MOD)%MOD<<endl;
    }
    return 0;
}