hdu 4507 数位dp

题目链接：hdu 4507

数位dp难度中等的一道题，判断的条件有三个，每个数字不为7可以在判断中排除，另外两个需要在递归基中判断

维护三个值：从pos位往后符合条件的数的数量num、这些数字的和sum，这些数字的平方和ssum

①num可以循环加得

②sum需要用用num再乘上一个当前数位所代表的值，即i*10^pos

③计算ssum时把sum和i*10^pos拆开完全平方分解

（i*10^pos+sum）^2=(i*10^pos)^2+sum^2(就是pos-1位置的ssum)+2*i*10^pos

判断情况时用%7的余数，相同的分为一种情况。

（函数传值写的int坑了自己好多发！）

#include<iostream>
#include<cstdio>
#include<vector>
#include<set>
#include<map>
#include<string.h>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#define LL long long
#define mod 1000000007
#define inf 0x3f3f3f3f
#define sqr(a) (a)*(a)
#define lan(a,b) memset(a,b,sizeof(a))

using namespace std;

struct node
{
    LL ssum,sum,num;

}dp[20][10][10];
int a[20];
LL p[20];

node pin(int pos,int mor,int smor,bool lim)
{
    if(pos==-1)
    {
       node tmp;
        tmp.num=(mor!=0 && smor!=0);
        tmp.sum=tmp.ssum=0;
        return tmp;

    }
    if(!lim&&dp[pos][mor][smor].num!=-1)
        return dp[pos][mor][smor];
    int up=lim?a[pos]:9;
    node ans;
    node tem;
    ans.num=ans.sum=ans.ssum=0;
    //printf("up=%d\n",up);
   for(int i=0;i<=up;i++)
    {
        if(i==7)continue;
        tem=pin(pos-1,(mor+i)%7,(smor*10+i)%7,lim&&i==up);
        ans.num+=tem.num;
        ans.num%=mod;
        ans.sum+=(tem.sum+ ((p[pos]*i)%mod)*tem.num%mod )%mod;
        ans.sum%=mod;

        ans.ssum+=(tem.ssum + ( (2*p[pos]*i)%mod )*tem.sum)%mod;
        ans.ssum%=mod;
        ans.ssum+=( (tem.num*p[pos])%mod*p[pos]%mod*i*i%mod );
        ans.ssum%=mod;
    }

    if(!lim) dp[pos][mor][smor]=ans;
    return ans;
  //  printf("%d %d %d=%d\n",pos,sta,lim,ans);
}

LL sov(LL n)
{
    int p=0;
    while(n)
    {
        a[p++]=n%10;
        n/=10;
    }
    return pin(p-1,0,0,1).ssum;
}

int main()
{
    LL n,m;
    //printf("%d\n",sov(100));
    int t;
    p[0]=1;
    for(int i=1;i<20;i++)
        p[i]=(p[i-1]*10)%mod;
for(int i=0;i<20;i++)
        for(int j=0;j<10;j++)
            for(int k=0;k<10;k++)
                dp[i][j][k].num=-1;

    scanf("%d",&t);
    while(t--)
    {
        scanf("%I64d%I64d",&n,&m);
        LL s1=sov(m);
        s1-=sov(n-1);
        s1=(s1%mod+mod)%mod;
        printf("%I64d\n",s1);
    }
    return 0;
}