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

Problem Description

　　单身!
　　依然单身！
　　吉哥依然单身！
　　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
  
  

  思路：dp[len][sum0][sum1] 表示长度为len对7取模为sum1，各位上的数字和取模为sum0有多少个满足的数；

唯一麻烦的是和7无关的数字的平方和，当我们进行位的搜索时需要用到以下方程；

对于一个数的平方和来讲：(pre*10^pos + next)^2= (pre*10^pos)^2+2*pre*10^pos*next +next^2   pre表示当前位，next表示下一位；

由于要记忆化，我们根据以上可以推出：

∑(Y + xi)^2 = ∑(Y^2 + 2 * Y * xi + xi^2) = n * Y^2 + 2 * Y * ∑xi + ∑ xi^2   

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;
const LL mod=1e9+7;
int b[20];
LL n,m,p[20];
struct node{
    LL cnt,sum,sqsum;
}dp[20][10][10];
node dfs(int len,int sum0,int sum1,int flag)
{
    if(len==0)
    {
        node str;
        str.cnt=(sum0!=0&&sum1!=0);
        str.sqsum=0;
        str.sum=0;
        return str;
    }
    if(!flag&&dp[len][sum0][sum1].cnt!=-1)
        return dp[len][sum0][sum1];
    int end=flag?b[len]:9;
    node now,next;
    now.cnt=now.sqsum=now.sum=0;
    for(int i=0;i<=end;i++)
    {
        if(i==7)
            continue;
        next=dfs(len-1,(sum0+i)%7,(sum1*10+i)%7,flag&&i==end); 
        now.cnt+=next.cnt%mod;  
        now.sum+=(next.sum+next.cnt*i%mod*p[len-1]%mod)%mod; //求出已经搜寻完的数的总值
        now.sum%=mod;
        now.sqsum+=(next.sqsum+2*next.sum%mod*i%mod*p[len-1]%mod)%mod;
        now.sqsum%=mod;
        now.sqsum+=i*i*p[len-1]%mod*p[len-1]%mod*next.cnt%mod;
        now.sqsum%=mod;
    }
    if(!flag)
        dp[len][sum0][sum1]=now;
    return now;
}
LL solve(__int64 n)
{
    int len=0;
    while(n)
    {
        b[++len]=n%10;
        n=n/10;
    }
    return dfs(len,0,0,1).sqsum;
}
int main()
{
    int t;
    p[0]=1;
    for(int i=1;i<20;i++)
        p[i]=p[i-1]*10%mod;
    for(int i=0; i<20; i++)
    {
        for(int j=0; j<10; j++)
        {
            for(int k=0; k<10; k++)
            {
                dp[i][j][k].cnt=-1;
            }
        }
    }
    scanf("%d",&t);
    while(t--)
    {
        scanf("%I64d%I64d",&n,&m);
        printf("%I64d\n",((solve(m)-solve(n-1)+mod)%mod));
    }
}