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

题目链接：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;
}