hdu 4507 数位dp 保存目标数平方和

如果一个整数符合下面3个条件之一，那么我们就说这个整数和7有关——
　　1、整数中某一位是7；
　　2、整数的每一位加起来的和是7的整数倍；
　　3、这个整数是7的整数倍；

　　现在问题来了：吉哥想知道在一定区间内和7无关的数字的平方和。

首先，第一个条件很好办，直接循环里面跳过7即可，第二个条件加一

维表示位数不断对7取模，第三个条件那么再加一维表示不断对这个数

对7取模，问题是保存平方数，当循环走到i*10^(pos-1)次方这里的时候

不是需要子问题加上去的么。把i*10^(pos-1)看成a, 然后(a+b)^2=a^2+b^2+2*a*b;

可以那咱们怎么知道加了多少次呢，所以每个dp的值不仅得保存x^的和，还得保存

有多少个数，以及sum（xi）；

#include <iostream>
#include <cstdio>
using namespace std;
typedef long long ll;
const ll Mod=1e9+7;
struct Node
{
    ll counter;
    ll sum;
    ll sqsum;
};
int up[25];
ll ten[20];
Node dp[25][10][10];
ll compute(ll n);
Node dfs(int pos,int add_mod,int mod,bool limit);
int main()
{
    int t;
    ten[0]=1;
    ll l,r;
    for(int i=1;i<=18;i++)
        ten[i]=ten[i-1]*10%Mod;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%I64d%I64d",&l,&r);
        printf("%I64d\n",(compute(r)-compute(l-1)+Mod)%Mod);
    }
    return 0;
}
ll compute(ll n)
{
    int cnt=0;
    while(n)
    {
        up[++cnt]=n%10;
        n/=10;
    }
    return dfs(cnt,0,0,true).sqsum;
}
Node dfs(int pos,int add_mod,int mod,bool limit)
{
    if(pos<=0)
    {
        Node temp;
        temp.counter=(add_mod&&mod);
        temp.sum=temp.sqsum=0;
        return temp;
    }
    if(!limit&&dp[pos][add_mod][mod].sum)
        return dp[pos][add_mod][mod];
    Node ans;
    ans.counter=ans.sum=ans.sqsum=0;
    int num=limit?up[pos]:9;
    for(int i=0;i<=num;i++)
    {
        if(i==7) continue;
        Node now=dfs(pos-1,(add_mod+i)%7,(mod*10+i)%7,limit&&i==num);
        ans.counter+=now.counter;
        ans.counter%=Mod;
        ans.sum+=(now.sum+ten[pos-1]*i%Mod*now.counter%Mod)%Mod;
        ans.sum%=Mod;
        ans.sqsum+=(now.sqsum+ten[pos-1]*i%Mod*ten[pos-1]%Mod*i%Mod*now.counter%Mod)%Mod;
        ans.sqsum%=Mod;
        ans.sqsum+=2*ten[pos-1]*i%Mod*now.sum%Mod;
        ans.sqsum%=Mod;
    }
    if(!limit)
        dp[pos][add_mod][mod]=ans;
    return ans;
}