吉哥系列故事――恨7不成妻 HDU - 4507 数位dp

时间:2023-03-09 03:16:27
吉哥系列故事――恨7不成妻 HDU - 4507  数位dp

思路  和普通的DP不一样的是 这里求的是满足条件的数的平方的和 而数位DP只跟数每位是什么密切相关  所以要开一个结构 (多加一个 数的和sum 和平方和qsum)存一下各个状态的和的情况

dp[pos][state1][state2].num  满足该状态的数有几个

dp[pos][state1][state2].sum 满足该条件的数的和是多少

dp[pos][state1][state2].qsum 满足该条件的数的平方的和是多少

详见注解 主要是状态转移是 和 和 平方和 的转移公式

∑(Y + xi)^2 = ∑(Y^2 + 2 * Y * xi + xi^2) = n * Y^2 + 2 * Y * ∑xi + ∑ xi^2  利用该公式 可以实现 状态转移

tips:先写好公式再疯狂% 不然容易乱

参考:https://blog.****.net/qq_37025443/article/details/78472991

#include <cstdio>
#include <cmath>
#include <algorithm>
#include<vector>
#include<iostream>
#include<cstring>
using namespace std;
const long long MOD=(1e9)+;
typedef long long ll;
ll sum;
int cnt;
ll a[];
int t[];
struct Node{
ll num,sum,qsum;
Node(ll a=,ll b=,ll c=):num(a),sum(b),qsum(c){ }
}dp[][][]; ll c[];
void init(){
c[]=;
for(int i=;i<=;i++)c[i]=(c[i-]*)%MOD;
} Node dfs(int pos,int state1,int state2,bool limit ){
if(pos==-){
return Node(state1&&state2,,);
}
if(!limit&&dp[pos][state1][state2].qsum!=)return dp[pos][state1][state2];
int up=limit?a[pos]:;
Node ans;
for(int i=;i<=up;i++){
// cout<<i<<endl;
if(i==)continue;
Node tmp=dfs(pos-,(state1+i)%,(state2*+i)%,limit&&i==up);
// cout<<111<<endl;
ans.num=(ans.num+tmp.num)%MOD;//满足的数求和
ans.sum=(ans.sum+(((i*c[pos])%MOD*tmp.num)%MOD+tmp.sum)%MOD)%MOD;//tmp.num*i*c[pos] 表示当前位pos的数的大小 乘以 一共有多少个满足条件数 tmp.sum表示pos-1的数的和
ans.qsum+=((tmp.qsum+(*i*c[pos])%MOD*tmp.sum)%MOD)%MOD;// 平方和公式实现状态转移 tmp.qsum 就是a^2 2*i*c[pos]*tmp.sum 就表示 2*a*b
ans.qsum%=MOD;
ans.qsum+=(((i*c[pos]*i)%MOD*c[pos])%MOD*tmp.num)%MOD;// i*i*c[pos]*c[pos] b^2 由上面的注解可以知道还需要求和 (n就是tmp.num)
ans.qsum%=MOD; }
if(!limit) dp[pos][state1][state2]=ans;
return ans;
} ll solve(ll x){
int pos=;
while(x){
a[pos++]=x%;
x/=;
}
return dfs(pos-,,,).qsum;
} int main()
{
ll a,b;
int t;
cin>>t;
int kase=;
init();
while(t--){
cin>>a>>b;
printf("%I64d\n",(solve(b)-solve(a-)+MOD)%MOD);
} return ;
}