hdu4507(数位dp,较复杂)

58 篇文章 1 订阅
19 篇文章 0 订阅
这题麻烦就是要求数的平方和。

我们先考虑求和的问题,一个区间,数位dp能在一些约束下计数,现在要这些数的和。其实组合数学搞搞就可以了:如 现在枚举的某一位pos,我统计了这一位枚举i的满足条件的个数cnt,其实只要算i对总和的贡献就可以了,对于一个数而已第pos位是i,那么对求和贡献就是i*10^pos,就是十进制的权值,然后有cnt个数都满足第pos位是i,最后sum=cnt*i*10^pos.原理就是这样平方和可以看做(a*10^pos+b)^2,a是你当前pos位要枚举的,b其实是个自问题,就是pos之后的位的贡献值,把这个平方展开就可以了!

具体见代码计算部分:(有的地方要乘上cnt的,有的地方不用)

#include <bits/stdc++.h>
#define ll long long 
using namespace std;
const int maxn = 20;
const ll MOD = 1e9 + 7;
struct CAL{
	ll sum;
	ll sqr;
	ll cnt;
	CAL() {sum =sqr=cnt=0;}
}dp[20][10][10];
ll a[maxn];
ll pow10[maxn];
void init()
{
	pow10[0] = 1;
	for(int i = 1;i <= 18;i++){
		pow10[i] = pow10[i - 1] * 10;
	}
	memset(dp,-1,sizeof dp);
}
ll pow2(ll x){
	return (x % MOD)* (x % MOD) % MOD;
}
CAL dfs(int pos, int sum, int val,bool lead,bool limit){
	if(pos == -1) {
		CAL p;
		if(sum != 0 && val != 0) p.cnt = 1;
		else p.cnt = 0;
		return p;
	}
	if(!limit && dp[pos][sum][val].cnt != -1) return dp[pos][sum][val];
	int up = limit ? pos[a] : 9;
	CAL ans ,temp;
	for(int i = 0;i <= up;i++){
		if(i == 7) continue;
		temp = dfs(pos - 1,(sum + i) % 7,(val * 10 + i) % 7,lead && (i == 0),limit && (i == up));
		ll tmp = ((temp.cnt * pow2(i * pow10[pos])) % MOD  + (2 * (i * pow10[pos]) % MOD * temp.sum) % MOD + temp.sqr) % MOD;
		ans.cnt = (ans.cnt + temp.cnt) % MOD;
		ans.sum = (ans.sum + temp.sum + ((i * pow10[pos]) % MOD * temp.cnt) % MOD)%MOD;
		ans.sqr = (ans.sqr + tmp) % MOD;
	}
	if(!limit) dp[pos][sum][val] = ans;
	return ans;
}
ll solve(ll x){
	int pos = 0;
	while(x > 0){
		a[pos++] = x % 10;
		x /= 10;
	}
	CAL ans = dfs(pos - 1,0,0,true,true);
	return ans.sqr;
}
int main()
{
	int kase;scanf("%d",&kase);
	init();
	while(kase--){
		ll aa,bb;
		scanf("%I64d%I64d",&aa,&bb);
		printf("%I64d\n",(solve(bb) - solve(aa - 1) + MOD) % MOD);
	}
	return 0;
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值