bzoj 3598 [Scoi2014]方伯伯的商场之旅——数位dp

时间:2023-03-10 03:09:15
bzoj 3598 [Scoi2014]方伯伯的商场之旅——数位dp

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3598

TJ:https://www.cnblogs.com/Zinn/p/9351218.html

核心:先确定一个中心,再考虑变动中心带来的好处。

用dfs实现很清楚。把填到这一位为止带来的代价写进参数里。但dfs的值表示的是把后面的位都填好带来的代价,所以要开long long。

这么一想好像也没啥???

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=,S=;
int K,dg[N];
ll l,r,dp[N][S];
ll dfs(int p,int s,bool fx)
{
if(!p)return s;
if(!fx&&dp[p][s]!=-)return dp[p][s];
int lm=K-;if(fx)lm=dg[p];
ll ret=;
for(int i=;i<=lm;i++)
ret+=dfs(p-,s+i*(p-),fx&(i==dg[p]));
if(!fx)dp[p][s]=ret;
return ret;
}
ll dfs(int p,int s,int zx,bool fx)
{
if(!p)return s>?s:;
if(!fx&&dp[p][s]!=-)return dp[p][s];
int lm=K-;if(fx)lm=dg[p];
ll ret=;
for(int i=;i<=lm;i++)
ret+=dfs(p-,p>zx?s+i:s-i,zx,fx&(i==dg[p]));
if(!fx)dp[p][s]=ret;
return ret;
}
ll calc(ll x)
{
int n=;
while(x)dg[++n]=x%K,x/=K;
memset(dp,-,sizeof dp);//
ll ret=dfs(n,,);
for(int i=;i<n;i++)
{
memset(dp,-,sizeof dp);
ret-=dfs(n,,i,);
}
return ret;
}
int main()
{
scanf("%lld%lld%d",&l,&r,&K);
printf("%lld\n",calc(r)-calc(l-));
return ;
}