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