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;
}