【数位dp+计算平方和+恶心又好】HDU - 4507 J - 吉哥系列故事――恨7不成妻

J - 吉哥系列故事――恨7不成妻   HDU - 4507

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

#include <iostream>
#include <cstdio>
#include <cstring>
#define ll long long
using namespace std;
const int mod = 1e9+7;
ll p[25];
int a[25];

struct status
{
    ll cnt, sum, sqsum;
    status() {cnt=-1, sum=sqsum=0;}  //初始化所有的dp[][][]
    status(ll cnt, ll sum, ll sqsum) : cnt(cnt), sum(sum), sqsum(sqsum) {} //可以通过status（1,0,0）这样的形式直接传递
}dp[20][10][10]; 
//dp[i][j][k]：长度为i的数的数位和为j，整体%7余数为k的数有几个(cnt)，他们之和(sum)，他们的平方和(sqsum)

status dfs(int pos, int prea, int preb, bool limit)
{
    if(pos == 0) return prea && preb ? status(1, 0, 0) : status(0, 0, 0); //除非满足数位和不为7的倍数，该数又不被7整除，cnt才等于1
    if(!limit && dp[pos][prea][preb].cnt!=-1) return dp[pos][prea][preb];
    status ans;
    ans.cnt = 0;
    int up = limit ? a[pos] : 9;
    for(int i=0; i<=up; i++)
    {
        if(i==7) continue;
        status next=dfs(pos-1, (prea+i)%7, (preb*10+i)%7, limit && a[pos] == i);
        ans.cnt+=next.cnt;
        ans.cnt%=mod;
        ans.sum+=next.sum+((p[pos]*i)%mod*next.cnt)%mod;  //此位有多少个增加了多少个i*p[pos]
        ans.sum%=mod;
        ans.sqsum+=next.sqsum+((2*p[pos]*i)%mod*next.sum)%mod;//(a+b)*(a+b) 加上2*a*b
        ans.sqsum%=mod;
        ans.sqsum+=(((p[pos]*i)%mod*p[pos])%mod*i)%mod*next.cnt;//加上b*b
        ans.sqsum%=mod;
    }
    if(!limit) dp[pos][prea][preb] = ans;
    return ans;
}

ll solve(ll x)
{
    int pos = 0;
    while(x)
    {
        a[++pos] = x%10;
        x/=10;
    }
    status ans = dfs(pos, 0, 0, 1);
    return ans.sqsum;
}

int main()
{
    int T;
    scanf("%d", &T);
    p[1]=1;
    for(int i=2; i<=20; i++)
    {
        p[i] = (p[i-1]*10)%mod;
    }
    while(T--)
    {
        ll n, m;
        scanf("%lld%lld", &n, &m);
        ll ans = solve(m);
        ans-=solve(n-1);//ans可能<0
        printf("%lld\n",(ans%mod+mod)%mod);
    }
    return 0;
}