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

题意

方伯伯有一天去参加一个商场举办的游戏。商场派了一些工作人员排成一行。每个人面前有几堆石子。说来也巧，位置在 i 的人面前的第 j 堆的石子的数量，刚好是 i 写成 K 进制后的第 j 位。
现在方伯伯要玩一个游戏，商场会给方伯伯两个整数 L,R。方伯伯要把位置在 [L, R] 中的每个人的石子都合并成一堆石子。每次操作，他可以选择一个人面前的两堆石子，将其中的一堆中的某些石子移动到另一堆，代价是移动的石子数量 * 移动的距离。商场承诺，方伯伯只要完成任务，就给他一些椰子，代价越小，给他的椰子越多。所以方伯伯很着急，想请你告诉他最少的代价是多少。
$1 < = L < = R < = 10^{15}, 2 < = K < = 20$

分析

先考虑只有一个数的话要怎么做。
如果我们把每颗石子拿出来，坐标是它的位置，若要把所有石子移到一起，显然全部移到中位数最优。
那么我们在数位dp之前可以先枚举所有石子的和，设为sum，那么就可以知道其中位数mid。
设f[i,j,0/1]表示从高位到低位做到第i位，前i位的石子数量和为j，是否卡着n的上界的总代价，g[i,j,0/1]表示反感。
接下来枚举下一位放什么。
若中位数恰好在下一位，则什么都不用做。
若中位数还未出现，那么之前的每一个石子肯定都要往后多移一位，把这里的贡献加上即可。
若中位数已经出现，那么后面的每一个石子肯定都要往前多移一位。
然后就做完了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define MAX(x,y) x=max(x,y)
using namespace std;

typedef long long LL;

int k,a[65];
LL f[65][305][2],g[65][305][2];

LL solve(LL n)
{
    int a1=0;LL ans=0;
    while (n) a[++a1]=n%k,n/=k;
    for (int sum=1;sum<a1*k;sum++)
    {
        int mid=(sum+1)/2;
        for (int i=1;i<=a1+1;i++)
            for (int j=0;j<=sum;j++)
                f[i][j][0]=f[i][j][1]=g[i][j][0]=g[i][j][1]=0;
        g[a1+1][0][1]=1;
        for (int i=a1+1;i>1;i--)
            for (int j=0;j<=sum;j++)
            {
                if (g[i][j][0])
                {
                    for (int l=0;l<k;l++)
                    {
                        if (j+l>sum) break;
                        g[i-1][j+l][0]+=g[i][j][0];
                        if (j+l<mid) f[i-1][j+l][0]+=f[i][j][0]+(LL)(j+l)*g[i][j][0];
                        else if (j>=mid) f[i-1][j+l][0]+=f[i][j][0]+(LL)(sum-j)*g[i][j][0];
                        else f[i-1][j+l][0]+=f[i][j][0];
                    }
                }
                if (g[i][j][1])
                {
                    for (int l=0;l<=a[i-1];l++)
                    {
                        if (j+l>sum) break;
                        g[i-1][j+l][l==a[i-1]]+=g[i][j][1];
                        if (j+l<mid) f[i-1][j+l][l==a[i-1]]+=f[i][j][1]+(LL)(j+l)*g[i][j][1];
                        else if (j>=mid) f[i-1][j+l][l==a[i-1]]+=f[i][j][1]+(LL)(sum-j)*g[i][j][1];
                        else f[i-1][j+l][l==a[i-1]]+=f[i][j][1];
                    }
                }
            }
        ans+=max(f[1][sum][0],(LL)0)+max(f[1][sum][1],(LL)0);
    }
    return ans;
}

int main()
{
    LL L,R;
    scanf("%lld%lld%d",&L,&R,&k);
    printf("%lld",solve(R)-solve(L-1));
    return 0;
}