2018.09.29【HDU4507】恨7不成妻（数位DP）

传送门

---

解析：

我数组开小，越界访问到了其他不该访问的位置，但是由于没有访问非法内存，没有$RE$，于是我就$WA$了十多次。。。

思路：

这种维护高次方和的问题肯定要从低次转移过来。

我们令$f[i][addremain][numremain]$表示之前高位的数字和$mod$ 7为$addremain$，前缀构成数字$mod$ 7为$numremain$的状态，是的这次又是只保存极限状态。

每一个$f$中的元素有三个属性，分别是$cnt,sum1,sum2$，分别表示满足该状态的数的0次方和（就是个数），一次方和，二次方和。

我们由低位向高位推导，状态转移为（假设我们的$ans$是第$pos$位的答案，$i$是我们枚举的下一位的数，$tmp$是我们下一状态的答案）
$$ans.cnt=ans.cnt+tmp.cnt$$$$res.sum1=res.sum1+tmp.sum1+10^{pos-1}\ast i\ast tmp.cnt$$
$$res.sum2+=tmp.sum2+//更低位答案的转移$$$$2\ast i\ast 10^{pos-1}\ast tmp.sum1+//枚举出的数与低位上各个数相乘$$$$tmp.cnt\ast (10^2{)}^{pos-1}\ast i^2//该位上的平方$$
就是这样，这道题就解决了。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc getchar
#define pc putchar
#define cs const

inline
ll getint(){
    re ll num;
    re char c;
    while(!isdigit(c=gc()));num=c^48;
    while(isdigit(c=gc()))num=(num<<1)+(num<<3)+(c^48);
    return num;
}

inline
void outint(ll a){
    static char ch[23];
    if(a==0)pc('0');
    while(a)ch[++ch[0]]=a-a/10*10,a/=10;
    while(ch[0])pc(ch[ch[0]--]^48);
}
cs ll mod=1000000007;

int len,mark[20];
ll bin[20];

struct node{
    ll cnt,sum1,sum2;
    node():cnt(0),sum1(0),sum2(0){}
}f[20][10][10];
//pos addremain numremain

inline
node dp(int pos,int addremain,int numremain,bool limit){
    if(!pos){
        node tmp;
        tmp.cnt=bool(addremain&&numremain);
        return tmp;
    }
    if(!limit&&(~f[pos][addremain][numremain].cnt))return f[pos][addremain][numremain];
    node res;
    int r=limit?mark[pos]:9;
    for(int re i=0;i<=r;++i){
        if(i==7)continue;
        node tmp=dp(pos-1,(addremain+i)%7,(numremain*10+i)%7,limit&&i==r);
        res.cnt=(res.cnt+tmp.cnt)%mod;
        res.sum1=(res.sum1+tmp.sum1+bin[pos-1]*i%mod*tmp.cnt%mod)%mod;
        res.sum2=((res.sum2+tmp.sum2)%mod+
                2*i*bin[pos-1]%mod*tmp.sum1%mod+
                tmp.cnt*bin[pos-1]%mod*bin[pos-1]%mod*i*i%mod)%mod;
    }
    if(!limit)f[pos][addremain][numremain]=res;
    return res;
}

inline
ll solve(ll a){
    len=0;
    do{
        mark[++len]=a-a/10*10;
        a/=10;
    }while(a);
    return dp(len,0,0,1).sum2;
}

int T;
ll l,r;
signed main(){
    bin[0]=1;
    for(int re i=1;i<20;++i)bin[i]=bin[i-1]*10%mod;
    memset(f,-1,sizeof f);
    T=getint();
    while(T--){
        l=getint(),r=getint();
        outint((solve(r)-solve(l-1)+mod)%mod);pc('\n');
    }
    return 0;
}