HDU4507 -- 数位DP

令 $f[i][j][k][p]$ 表示考虑前 $i$ 位，数位值之和模 $7$ 的值为 $j$ ，数值模 $7$ 的值为 $k$ , 前 $i$ 位是否取满时的平方和。

$(a+b)^2=a^2+2ab+b^2$

记一下方案数、和、平方和就可以转移了。

时间复杂度 $O(T*18*7*7*9)$ 。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 20
#define M 1000000007
#define ll long long
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void Read(ll& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
ll x,y;
int i,j,k,n,m,P[N<<1],P2[N<<1],T,p,q,j1,k1,p1,q1;
int l,a[N];
int f[2][7][7][2][3];
bool b;
inline int Calc(ll x){
    if(!x)return 0;
    l=0;
    while(x)a[++l]=x%10,x/=10;
    for(i=1;i<=l/2;i++)swap(a[i],a[l-i+1]);
    memset(f[1],0,sizeof(f[1]));
    b=0;f[1][0][0][1][0]=1;
    for(i=1;i<=l;i++,b^=1){
        memset(f[b],0,sizeof(f[b]));
        for(j=0;j<7;j++)
        for(k=0;k<7;k++)
        for(p=0;p<2;p++)
        for(q=0;q<=9;q++)
        if(q!=7&&(!p||q<=a[i])&&f[b^1][j][k][p][0]){
            j1=(j+q)%7;k1=(k+q*P2[l-i]%7)%7;p1=(p&&q==a[i]);
            (f[b][j1][k1][p1][0]+=f[b^1][j][k][p][0])%=M;
            (f[b][j1][k1][p1][1]+=(f[b^1][j][k][p][1]+1ll*q*P[l-i]%M*f[b^1][j][k][p][0]%M)%M)%=M;
            (f[b][j1][k1][p1][2]+=((f[b^1][j][k][p][2]+2ll*f[b^1][j][k][p][1]*q%M*P[l-i]%M)%M+1ll*q*q%M*P[l-i<<1]%M*f[b^1][j][k][p][0]%M)%M)%=M;
        }
    }
    int Ans=0;b^=1;
    for(i=1;i<7;i++)
    for(j=1;j<7;j++)
    for(p=0;p<2;p++)
    (Ans+=f[b][i][j][p][2])%=M;
    return Ans;
}
int main(){
    scanf("%d",&T);
    for(P[0]=P2[0]=i=1;i<40;i++)P[i]=1ll*P[i-1]*10%M,P2[i]=P2[i-1]*10%7;
    while(T--){
        Read(x);Read(y);
        printf("%d\n",(Calc(y)-Calc(x-1)+M)%M);
    }
    return 0;
}