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

唉,以前就做过了,重做了一遍。居然花了4、5个小时,才做出来。各种惨不忍睹的错误o(╯□╰)o

其实,无论是求和,还是求平方和,都是较裸的数位DP。求和还是比较简单的,这里就谈谈怎么求平方和。

比如,我们要求ABCX*ABCX。由ABCX==ABC0 + X,整理有,ABCX*ABCX == ABC0*ABC0 + X*X + 2*ABC0*X

我们用dp[i][0/1][p][q]表示,从高位开始,已经填好了i个数字,当前是否(1/0)将这前i位置满,每位和除以7余p,前i位除以7余q,同时,前i位中,不能有7,这个状态下,有多少个数字。当然,sum[ ][ ][ ] []则表示这状态下,数字的和,sum2[ ][ ][ ][ ]则是平方和。

当我们要计算sum2[k][ ][ ][ ]的时候,按我们ABCX那个例子知,可以由dp[k-1][ ][ ][ ]和sum[k-1][ ][ ][ ]推出来。


//#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<cctype>
#include<string>
#include<algorithm>
#include<iostream>
#include<ctime>
#include<map>
#include<set>
using namespace std;
#define MP(x,y) make_pair((x),(y))
#define PB(x) push_back(x)
typedef __int64 LL;
//typedef unsigned __int64 ULL;
/* ****************** */
const int INF=1000111222;
const double INFF=1e200;
const double eps=1e-8;
const LL mod=1000000007;
const int NN=1100;
const int MM=10010;
/* ****************** */

int a[22];
LL pow10[22];
LL dp[22][2][8][8];
LL sum[22][2][8][8];
LL sum2[22][2][8][8];

void calc_cnt(int tol)
{
    memset(dp,0,sizeof(dp));
    int i,j,k,k1;
    int nj,nk;
    dp[0][1][0][0]=1;
    for(i=0;i<tol;i++)
    {
        for(j=0;j<7;j++)
            for(k=0;k<7;k++)
            {
                if(dp[i][0][j][k])
                {
                    for(k1=0;k1<10;k1++)
                    {
                        if(k1==7)
                            continue;
                        nj=(j+k1)%7;
                        nk=(k*10+k1)%7;
                        dp[i+1][0][nj][nk]+=dp[i][0][j][k];
                        if(dp[i+1][0][nj][nk]>=mod)
                            dp[i+1][0][nj][nk]-=mod;
                    }
                }
                if(dp[i][1][j][k])
                {
                    for(k1=0;k1<a[i+1];k1++)
                    {
                        if(k1==7)
                            continue;
                        nj=(j+k1)%7;
                        nk=(k*10+k1)%7;
                        dp[i+1][0][nj][nk]+=dp[i][1][j][k];
                        if(dp[i+1][0][nj][nk]>=mod)
                            dp[i+1][0][nj][nk]-=mod;
                    }
                    if(k1!=7)
                    {
                        nj=(j+k1)%7;
                        nk=(k*10+k1)%7;
                        dp[i+1][1][nj][nk]+=dp[i][1][j][k];
                        if(dp[i+1][1][nj][nk]>=mod)
                            dp[i+1][1][nj][nk]-=mod;
                    }
                }
            }
    }
/*
    puts("fuck!");
    for(i=0;i<=tol;i++)
    {
        for(j=0;j<7;j++)
        {
            for(k=0;k<7;k++)
            {
                cout<<dp[i][0][j][k]<<" ";
            }
            cout<<endl;
        }
        puts("-----");
    }
    puts("fuck!!!!!!!!!");
    */
}

void calc_sum(int tol)
{
    memset(sum,0,sizeof(sum));
    int i,j,k,k1;
    int nj,nk;
    LL t;
    for(i=0;i<tol;i++)
    {
        for(j=0;j<7;j++)
            for(k=0;k<7;k++)
            {
                for(k1=0;k1<10;k1++)
                {
                    if(k1==7)
                        continue;
                    nj=(j+k1)%7;
                    nk=(k*10+k1)%7;

                    t=(pow10[tol-1-i]*k1%mod)*dp[i][0][j][k];
                    t%=mod;
                    sum[i+1][0][nj][nk]+=t;
                    if(sum[i+1][0][nj][nk]>=mod)
                        sum[i+1][0][nj][nk]-=mod;

                    t=sum[i][0][j][k];
                    sum[i+1][0][nj][nk]+=t;
                    if(sum[i+1][0][nj][nk]>=mod)
                        sum[i+1][0][nj][nk]-=mod;

                }
                for(k1=0;k1<a[i+1];k1++)
                {
                    if(k1==7)
                        continue;
                    nj=(j+k1)%7;
                    nk=(k*10+k1)%7;

                    t=(pow10[tol-1-i]*k1%mod)*dp[i][1][j][k];
                    t%=mod;
                    sum[i+1][0][nj][nk]+=t;
                    if(sum[i+1][0][nj][nk]>=mod)
                        sum[i+1][0][nj][nk]-=mod;

                    t=sum[i][1][j][k];
                    sum[i+1][0][nj][nk]+=t;
                    if(sum[i+1][0][nj][nk]>=mod)
                        sum[i+1][0][nj][nk]-=mod;
                }
                if(k1!=7)
                {
                    nj=(j+k1)%7;
                    nk=(k*10+k1)%7;

                    t=(pow10[tol-1-i]*k1%mod)*dp[i][1][j][k];
                    t%=mod;
                    sum[i+1][1][nj][nk]+=t;
                    if(sum[i+1][1][nj][nk]>=mod)
                        sum[i+1][1][nj][nk]-=mod;

                    t=sum[i][1][j][k];
                    sum[i+1][1][nj][nk]+=t;
                    if(sum[i+1][1][nj][nk]>=mod)
                        sum[i+1][1][nj][nk]-=mod;
                }
            }
    }
/*
    puts("sum!");
    for(i=0;i<=tol;i++)
    {
        for(j=0;j<7;j++)
        {
            for(k=0;k<7;k++)
            {
                cout<<sum[i][1][j][k]<<" ";
            }
            cout<<endl;
        }
        puts("-----");
    }
    puts("sum!!!!!!!!!");
*/
}

void calc_sum2(int tol)
{
    memset(sum2,0,sizeof(sum2));
    int i,j,k,k1;
    int nj,nk;
    LL t;
    for(i=0;i<tol;i++)
    {
        for(j=0;j<7;j++)
            for(k=0;k<7;k++)
            {
                for(k1=0;k1<10;k1++)
                {
                    if(k1==7)
                        continue;
                    nj=(j+k1)%7;
                    nk=(k*10+k1)%7;

                    sum2[i+1][0][nj][nk]+=sum2[i][0][j][k];
                    if(sum2[i+1][0][nj][nk]>=mod)
                        sum2[i+1][0][nj][nk]-=mod;

                    t=pow10[tol-1-i]*k1%mod;
                    t*=sum[i][0][j][k]*2;
                    t%=mod;
                    sum2[i+1][0][nj][nk]+=t;
                    if(sum2[i+1][0][nj][nk]>=mod)
                        sum2[i+1][0][nj][nk]-=mod;

                    t=pow10[tol-1-i]*k1%mod;
                    t=t*t%mod;
                    t=t*dp[i][0][j][k];
                    t%=mod;
                    sum2[i+1][0][nj][nk]+=t;
                    if(sum2[i+1][0][nj][nk]>=mod)
                        sum2[i+1][0][nj][nk]-=mod;
                }
                for(k1=0;k1<a[i+1];k1++)
                {
                    if(k1==7)
                        continue;
                    nj=(j+k1)%7;
                    nk=(k*10+k1)%7;

                    sum2[i+1][0][nj][nk]+=sum2[i][1][j][k];
                    if(sum2[i+1][0][nj][nk]>=mod)
                        sum2[i+1][0][nj][nk]-=mod;

                    t=pow10[tol-1-i]*k1%mod;
                    t*=sum[i][1][j][k]*2;
                    t%=mod;
                    sum2[i+1][0][nj][nk]+=t;
                    if(sum2[i+1][0][nj][nk]>=mod)
                        sum2[i+1][0][nj][nk]-=mod;

                    t=pow10[tol-1-i]*k1%mod;
                    t=t*t%mod;
                    t=t*dp[i][1][j][k];
                    t%=mod;
                    sum2[i+1][0][nj][nk]+=t;
                    if(sum2[i+1][0][nj][nk]>=mod)
                        sum2[i+1][0][nj][nk]-=mod;
                }
                if(k1!=7)
                {
                    nj=(j+k1)%7;
                    nk=(k*10+k1)%7;

                    sum2[i+1][1][nj][nk]+=sum2[i][1][j][k];
                    if(sum2[i+1][1][nj][nk]>=mod)
                        sum2[i+1][1][nj][nk]-=mod;

                    t=pow10[tol-1-i]*k1%mod;
                    t*=sum[i][1][j][k]*2;
                    t%=mod;
                    sum2[i+1][1][nj][nk]+=t;
                    if(sum2[i+1][1][nj][nk]>=mod)
                        sum2[i+1][1][nj][nk]-=mod;

                    t=pow10[tol-1-i]*k1%mod;
                    t=t*t%mod;
                    t=t*dp[i][1][j][k];
                    t%=mod;
                    sum2[i+1][1][nj][nk]+=t;
                    if(sum2[i+1][1][nj][nk]>=mod)
                        sum2[i+1][1][nj][nk]-=mod;
                }
            }
    }
}

LL solve(LL x)
{
    int i,j,tol=0;
    LL ans=0;
    do
    {
        a[++tol]=x%10;
        x=x/10;
    }while(x);
    for(i=1;i<=tol/2;i++)
        swap(a[i],a[tol+1-i]);

  //  puts("fen");
  //  for(i=1;i<=tol;i++)
  //      printf("%d%c",a[i],i==tol?'\n':' ');

    pow10[0]=1;
    for(i=1;i<22;i++)
    {
        pow10[i]=pow10[i-1]*10;
        pow10[i]%=mod;
    }

    calc_cnt(tol);
    calc_sum(tol);
    calc_sum2(tol);

    for(i=1;i<7;i++)
        for(j=1;j<7;j++)
        {
            ans+=sum2[tol][0][i][j];
            if(ans>=mod)
                ans-=mod;
            ans+=sum2[tol][1][i][j];
            if(ans>=mod)
                ans-=mod;
        }
    return ans;
}

int main()
{
   // freopen("E:\\in.txt","r",stdin);
  //  freopen("E:\\no.txt","w",stdout);

    int cas;
    LL s,e,ans;
    scanf("%d",&cas);
    while(cas--)
    {
        cin>>s>>e;

     //   cout<<"s=="<<solve(s-1)<<endl;
    //    cout<<"e=="<<solve(e)<<endl;

        ans=solve(e)-solve(s-1);
        if(ans<0)
            ans+=mod;
        cout<<ans<<endl;
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值