HDU4507 吉哥系列故事——恨7不成妻

思路:数位dp,维护三个值,与7无关的数的个数cnt,这cnt个数的和sum,这cnt个数的平方和sqsum

cnt好算,sum则是根据cnt来计算,例如当前状态是这样1234***(确定了前四位,当前确定的是第四位,‘4’),由于递归,在前四位确定的情况下会计算出后三位的共有cnt状态符合使得这个7位数满足“与7无关” 。则算sum时,第4位上的“4"应加cnt次。

sqsum则可以通过展开(a + b)^2= a^2+2*a*b +b^2来利用sum和sqsum计算新的sqsum。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
using namespace std;
typedef long long LL;
const LL mod=1e9+7;
int t,bit[22];
LL L,R,fac[22];
struct node
{
    LL cnt,sum,sqsum;
}dp[22][10][10];
node dfs(int len,int sum1,int sum2,bool bound)
{
    node tmp;
    if(len<0)
    {
        if(sum1&&sum2) tmp.cnt=1;
        else tmp.cnt=0;
        tmp.sum=tmp.sqsum=0;
        return tmp;
    }
    if(!bound&&dp[len][sum1][sum2].cnt!=-1) return dp[len][sum1][sum2];
    node ans;
    ans.cnt=ans.sum=ans.sqsum=0;
    int end=bound?bit[len]:9;
    for(int i=0;i<=end;i++)
    {
        if(i==7) continue;
        tmp=dfs(len-1,(sum1+i)%7,(sum2*10+i)%7,bound&&i==end);
        ans.cnt=(ans.cnt+tmp.cnt)%mod;
        ans.sum=(ans.sum+tmp.sum+i*fac[len]%mod*tmp.cnt%mod)%mod;
        ans.sqsum=(ans.sqsum+tmp.sqsum+2*i*fac[len]%mod*tmp.sum%mod+tmp.cnt*i*fac[len]%mod*i%mod*fac[len]%mod)%mod;
    }
    if(!bound) dp[len][sum1][sum2]=ans;
    return ans;
}
LL solve(LL num)
{
    int len=0;
    while(num)
    {
        bit[len++]=num%10;
        num/=10;
    }
    return dfs(len-1,0,0,true).sqsum;
}
int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d",&t);
    fac[0]=1;
    for(int i=1;i<20;i++)
        fac[i]=(fac[i-1]*10)%mod;
    memset(dp,-1,sizeof(dp));
    while(t--)
    {
        scanf("%I64d%I64d",&L,&R);
        printf("%I64d\n",((solve(R)-solve(L-1))+mod)%mod);
    }
    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值