吉哥系列故事——恨7不成妻 （数位dp）

https://cn.vjudge.net/contest/312854#problem/E

单身! 
　　依然单身！ 
　　吉哥依然单身！ 
　　DS级码农吉哥依然单身！ 
　　所以，他生平最恨情人节，不管是214还是77，他都讨厌！ 
　　 
　　吉哥观察了214和77这两个数，发现： 
　　2+1+4=7 
　　7+7=7*2 
　　77=7*11 
　　最终，他发现原来这一切归根到底都是因为和7有关！所以，他现在甚至讨厌一切和7有关的数！ 

　　什么样的数和7有关呢？ 

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

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

Input

输入数据的第一行是case数T(1 <= T <= 50)，然后接下来的T行表示T个case;每个case在一行内包含两个正整数L, R(1 <= L <= R <= 10^18)。 

Output

请计算[L,R]中和7无关的数字的平方和，并将结果对10^9 + 7 求模后输出。

Sample Input

3
1 9
10 11
17 17

Sample Output

236
221
0

如果只是求数目的话那么很easy，但是题目让求平方和

假设第pos位的数是i,那么它实际代表的数是i*10^(pos-1),不妨设为f

设pos-1位到1位组成的数字是x

（f+x）^2=f^2+x^2+2*f*x

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
struct node{
	ll num;
	ll sum1;
	ll sum2;
}dp[20][7][7];
int a[20];
ll b[20];
void init(){
	for(int i=0;i<20;i++){
		for(int j=0;j<7;j++){
			for(int k=0;k<7;k++){
				dp[i][j][k].num=-1;
			}
		}
	}
	b[0]=1;
	for(int i=1;i<19;i++)
	b[i]=10*b[i-1];
}
node dfs(int pos,int sta1,int sta2,bool limit){
	if(pos==0){
		node tmp;
		tmp.num=(sta1&&sta2);
		tmp.sum1=tmp.sum2=0;
		return tmp;
	}
	if(!limit&&dp[pos][sta1][sta2].num!=-1) return dp[pos][sta1][sta2];
	int up=limit?a[pos]:9;
	node ans,tmp;
	ans.num=ans.sum1=ans.sum2=0;
	for(int i=0;i<=up;i++){
		if(i==7) continue;
		tmp=dfs(pos-1,(sta1+i)%7,(sta2*10+i)%7,limit&&i==up);
		ans.num+=tmp.num;
		ans.num%=mod;
		ll f=i*b[pos-1]%mod;
		ans.sum1+=(tmp.sum1+f*tmp.num%mod)%mod;
		ans.sum1%=mod;
		ans.sum2+=(tmp.sum2+f*f%mod*tmp.num%mod+2*f*tmp.sum1%mod)%mod;
		ans.sum2%=mod;
	}
	if(!limit) dp[pos][sta1][sta2]=ans;
	return ans;
}
ll solve(ll x){
	int pos=0;
	while(x){
		a[++pos]=x%10;
		x/=10;
	}
	return dfs(pos,0,0,1).sum2;
}
int main(){
	int T;
	scanf("%d",&T);
	init();
	while(T--){
		ll a,b;
		scanf("%lld%lld",&a,&b);
		printf("%lld\n",(solve(b)-solve(a-1)%mod+mod)%mod);
	}
	return 0;
}