HDU-4507 吉哥系列故事——恨7不成妻

数位DP，一般的是让人求某种数字的个数，这个要求求这些数字的平方和

刚开始我以为递归到结尾把数字平方加起来就好了，结果是WA（数位DP并不能遍历每一个数字，所以这样加肯定是错的）

这样的话需要用到一些公式，帮助求平方和： https://www.cnblogs.com/kuangbin/archive/2013/05/01/3053233.html 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
const int N=20;
const ll mod=1e9+7;
ll p[N];
struct node
{
	ll cnt;
	ll sum;
	ll sqsum;
}d[N][10][10];
int dis[N];
node dp(int len,int f,int g,bool flag)
{
	if(len==-1)
	{
		node tmp;
		if(f!=0&&g!=0) tmp.cnt=1;
		else tmp.cnt=0;
		tmp.sum=tmp.sqsum=0;
		return tmp;
	}
	if(!flag&&d[len][f][g].cnt!=-1) return d[len][f][g];
	node ans;
	node u;
	ans.cnt=ans.sum=ans.sqsum=0;
	int end=flag?dis[len]:9;
	for(int i=0;i<=end;i++)
	{
		if(i==7) continue;
		u=dp(len-1,(f+i)%7,(g*10+i)%7,flag&&(i==end));
		ans.cnt+=u.cnt;
		ans.cnt%=mod;
		ans.sum+=(u.sum+((i*p[len])%mod)*u.cnt)%mod;
		ans.sum%=mod;
		ans.sqsum+=(u.sqsum+((2*p[len]*i)%mod)*u.sum)%mod;
                ans.sqsum%=mod;
                ans.sqsum+=((u.cnt*p[len])%mod*p[len]%mod*i*i%mod);
                ans.sqsum%=mod;
	}
	if(!flag) d[len][f][g]=ans;
	return ans;
}
ll solve(ll n)
{
	int t=0;
	while(n)
	{
		dis[t++]=n%10;
		n/=10;
	}
	return dp(t-1,0,0,true).sqsum;
}
int main()
{
	int T;
	scanf("%d",&T);
	memset(d,-1,sizeof(d));
	p[0]=1;
	for(int i=1;i<N;i++)
		p[i]=(p[i-1]*10)%mod;
	ll a,b;
	while(T--)
	{
		scanf("%lld%lld",&a,&b);
		printf("%lld\n",((solve(b)-solve(a-1))%mod+mod)%mod);
	}
	return 0;
}