题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4507
数位DP专题:http://blog.csdn.net/chy20142109/article/details/50930004
题意:统计在[L,R]区间内所有与7无关数字的平方和。
什么样的数和7有关呢?
如果一个整数符合下面3个条件之一,那么我们就说这个整数和7有关——
1、整数中某一位是7;
2、整数的每一位加起来的和是7的整数倍;
3、这个整数是7的整数倍;
思路:如果是统计区间内数的个数就比较简单,那么我们来考虑一下怎么去统计平方和。
假设当前枚举第k位的值是i,那么我们需要计算以i开头的数字的平方和。
平方和公式展开(i+y)² = i² + 2 * i * y + y² 如果有t个以i开头的数的话,就是n * i² + 2 * i * ∑y + ∑(y²) 这里的i表示i*10^(k-1)
所以我们在一层中去维护三个值,一个出现次数,一个当前层的所有满足数的和,还有一个平方和。每次通过下一次dfs返回的值来更新当前层的值,再返回
到上一层。注意乘的时候不要一次乘太多再取模,防止溢出。
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <utility>
using namespace std;
#define rep(i,j,k) for (int i=j;i<=k;i++)
#define Rrep(i,j,k) for (int i=j;i>=k;i--)
#define Clean(x,y) memset(x,y,sizeof(x))
#define LL long long
#define ULL unsigned long long
#define inf 0x7fffffff
const long long mod=1000000007LL;
struct node
{
LL t;
LL sum;
LL squre_sum;
node(LL a = 0,LL b = 0,LL c = 0):t(a),sum(b),squre_sum(c){}
};
node dp[20][10][10];
int bit[20];
LL P[20];
int T;
LL L,R;
node dfs(int k , int yu , int sum , bool flag)
{
if ( k == 0 )
{
node ans;
ans.t = ( sum!=0 && yu !=0 );
ans.squre_sum = ans.sum = 0;
return ans;
}
if ( !flag && dp[k][yu][sum].t != -1 ) return dp[k][yu][sum];
node ans;
node temp;
int ed = flag?bit[k]:9;
rep(i,0,ed)
{
if ( i == 7 ) continue;
temp = dfs( k-1 , (yu*10+i)%7 , (sum+i)%7 , flag && (i == ed) );
ans.t = ( ans.t + temp.t )% mod;
ans.sum = ( ans.sum + temp.sum + (( i * P[k] )% mod * (temp.t % mod) )%mod ) % mod;
ans.squre_sum = ( ans.squre_sum + temp.squre_sum + ( ( ( i * i * P[k] ) % mod * P[k] )% mod *temp.t) % mod + ((2 * i * P[k]) %
mod * temp.sum )% mod )%mod ;
}
if ( !flag ) dp[k][yu][sum] = ans;
return ans;
}
LL cal(LL x)
{
int len = 0;
while(x)
{
bit[++len] = x % 10;
x/=10;
}
node ans = dfs(len,0,0,1);
return ans.squre_sum;
}
int main()
{
P[1] = 1;
rep(i,2,19) P[i] = ( 10 * P[i-1] ) % mod;
Clean(dp,-1);
cin>>T;
while(T--)
{
scanf("%I64d %I64d",&L,&R);
// cout<<cal(R)<<endl;
//cout<<cal(L-1)<<endl;
printf("%I64d\n", ( (cal(R)-cal(L-1)) + mod )%mod );
}
return 0;
}