hdu 4507 数位dp（求和，求平方和）

标签：

http://acm.hdu.edu.cn/showproblem.php?pid=4507

3
1 9
10 11
17 17

236
221
0

/***
hdu 4507  数位dp（求和，求平方和）
解题思路：dp[len][sum1][sum2] 表示长度为len对7取模为sum1，各位上的数字和为sum2有多少个满足的数
          一个是与7无关的数的个数，就是简单的数位DP了，很常规。
          第二个与7无关的数的和的维护需要用到第一个个数。
          处理到第pos个数位时，加上i*10^pos * 后面的个数
          第三个的维护需要用到前面两个
          (pre*10^pos + next)^2= (pre*10^pos)^2+2*pre*10^pos*next +next^2
*/
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <iostream>
using namespace std;
typedef long long LL;
const LL mod=1e9+7;
LL l,r,p[25];
int bit[25];

struct node
{
    LL cnt,sum,sqsum;
} dp[25][10][10];

node dfs(int len,int sum1,int sum2,int flag)
{
    if(len<0)
    {
        node tmp;
        tmp.cnt=(sum1!=0&&sum2!=0);
        tmp.sum=tmp.sqsum=0;
        return tmp;
    }
    if(flag==0&&dp[len][sum1][sum2].cnt!=-1)return dp[len][sum1][sum2];
    node ans,tmp;
    int end=flag?bit[len]:9;
    ans.cnt=ans.sqsum=ans.sum=0;
    for(int i=0; i<=end; i++)
    {
        if(i==7)continue;
        tmp=dfs(len-1,(sum1+i)%7,(sum2*10+i)%7,flag&&i==end);
        ans.cnt+=tmp.cnt;
        ans.cnt%=mod;
        ans.sum+=(tmp.sum+i*p[len]%mod*tmp.cnt%mod)%mod;
        ans.sum%=mod;
        ans.sqsum+=(tmp.sqsum+2*p[len]*i%mod*tmp.sum%mod)%mod;
        ans.sqsum%=mod;
        ans.sqsum+=(tmp.cnt*p[len]%mod*p[len]%mod*i*i%mod);
        ans.sqsum%=mod;
    }
    if(flag==0)dp[len][sum1][sum2]=ans;
    return ans;
}
LL solve(LL n)
{
    int len=0;
    while(n)
    {
        bit[len++]=n%10;
        n/=10;
    }
    return dfs(len-1,0,0,1).sqsum;
}
int main()
{
    p[0]=1;
    for(int i=1; i<20; i++)
        p[i]=(p[i-1]*10)%mod;
    for(int i=0; i<25; i++)
    {
        for(int j=0; j<10; j++)
        {
            for(int k=0; k<10; k++)
            {
                dp[i][j][k].cnt=-1;
            }
        }
    }
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%I64d%I64d",&l,&r);
        printf("%I64d\n",((solve(r)-solve(l-1))%mod+mod)%mod);
    }
    return 0;
}

hdu 4507 数位dp（求和，求平方和）

标签：

原文地址：http://blog.csdn.net/lvshubao1314/article/details/46348983