【bzoj 3598】: [Scoi2014]方伯伯的商场之旅

http://www.lydsy.com/JudgeOnline/problem.php?id=3598

DP啊DP

如果枚举mid的话会很麻烦。。。可能还会爆空间

改成转移一切都不是问题~

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <cmath>
#include <algorithm>
using namespace std;
//////////////////////////////////////////////////
#define rep(i,l,r) for(int i=(l),_=(r);i<=_;i++)
#define per(i,r,l) for(int i=(r),_=(l);i>=_;i--)
#define INE(i,u) for(int i=head[u];~i=i=e[i].next)
#define MS(arr,x) memset(arr,x,sizeof(arr))
#define LL long long
#define X first
#define Y second
typedef pair<LL,LL>pll;
//////////////////////////////////////////////////
LL L,R,k;
int bit[60];
pll dp[60];
LL dp2[60][3000][60];
//////////////////////////////////////////////////
pll dfs(int pos,bool limit)
{
if(!pos) return pll(1,0);
if(!limit&&~dp[pos].X) return dp[pos];
pll res(0,0);
int upp=limit?bit[pos]:k-1;
rep(i,0,upp)
{
pll ret=dfs(pos-1,limit&&i==upp);
res.X+=ret.X;
res.Y+=ret.Y+ret.X*i*(pos-1);
}
if(!limit) dp[pos]=res;
return res;
}
LL dfs2(int pos,int c,int mid,bool limit)
{
if(!pos) return c;
if(!limit&&~dp2[pos][c][mid]) return dp2[pos][c][mid];
int upp=limit?bit[pos]:k-1;
LL res=0;
rep(i,0,upp)
{
int delta=i;
if(pos<mid) delta=-delta;
int nc=c+delta;
if(nc<0) continue;
res+=dfs2(pos-1,nc,mid,limit&&i==upp);
}
if(!limit) dp2[pos][c][mid]=res;
return res;
}
LL solve(LL n)
{
for(*bit=0;n;n/=k) bit[++*bit]=n%k;
LL res=dfs(*bit,1).Y;
rep(i,2,*bit)
res-=dfs2(*bit,0,i,1);
return res;
}
//////////////////////////////////////////////////
void input()
{
MS(dp,-1); MS(dp2,-1);
cin>>L>>R>>k;
}
void solve()
{
cout<<solve(R)-solve(L-1)<<endl;
}
int main()
{
freopen("shop.in","r",stdin); freopen("shop.out","w",stdout);
input(),solve();
return 0;
}