BZOJ3598 [Scoi2014]方伯伯的商场之旅

Description

方伯伯有一天去参加一个商场举办的游戏。商场派了一些工作人员排成一行。每个人面前有几堆石子。说来也巧，位置在 i 的人面前的第 j 堆的石子的数量，刚好是 i 写成 K 进制后的第 j 位。

现在方伯伯要玩一个游戏，商场会给方伯伯两个整数 L,R。方伯伯要把位置在 [L, R] 中的每个人的石子都合并成一堆石子。每次操作，他可以选择一个人面前的两堆石子，将其中的一堆中的某些石子移动到另一堆，代价是移动的石子数量 * 移动的距离。商场承诺，方伯伯只要完成任务，就给他一些椰子，代价越小，给他的椰子越多。所以方伯伯很着急，想请你告诉他最少的代价是多少。

例如：10 进制下的位置在 12312 的人，合并石子的最少代价为：

1 * 2 + 2 * 1 + 3 * 0 + 1 * 1 + 2 * 2 = 9

即把所有的石子都合并在第三堆

Input

输入仅有 1 行，包含 3 个用空格分隔的整数 L，R，K，表示商场给方伯伯的 2 个整数，以及进制数

Output

 输出仅有 1 行，包含 1 个整数，表示最少的代价。

Sample Input

3 8 3

Sample Output

5

HINT

 1 < =  L < =  R < =  10^15, 2 < =  K < =  20

题解：

       如果一个数所有数位和为sum，那么一定是将所有数字移动到sum/2这个位置。根据这个思路，枚举这个位置mid和mid上的数字v。

       我们设mid左边的数位和为S1，mid右边的数位和为S2。如果将mid往左移一位，答案改变S2+mid-S1；将mid往右移一位，答案改变S1+mid-S2。两者都要大于等于0，于是解得 -v <=S1-S2 <= v ，在DP时为了去重，我们将随意一边改为开区间，即：当S1-S2∈(-v,v]时，我们认为是合法的。

       然后就可以糊DP了。一开始算错复杂度糊了个非常假的做法：枚举mid和v，用f[i][j][k][0/1]表示前i位，和为j(即S1-S2)，代价和为k，无/有高位限制的方案数，然后答案就是∑k*f[i][j][k][0/1]。

       T了之后想到k这一维是没有必要的：如果知道了方案数，直接将这一位上的代价乘以方案数就可以得到这一位产生的费用，根本不需要记录前面位数产生的费用。于是用f[i][j][0/1]表示前i位，和为j，无/有高位限制的费用，g[i][j][0/1]表示方案数再DP即可。

       代码大改了好几次非常混乱……

#pragma GCC optimize(3) 
#include<bits/stdc++.h>
#define LL long long
using namespace std;

int a[50],mid,midv,cnt,base;
LL L,R,f[59][505][2],g[59][505][2];

LL dp(int x,int y,int t) {
	if (x==cnt+1) return 0;
	return f[x][y][t];
}

LL Solve(int x,int y,int t) {
	if (x==cnt+1) {
		int yy=y-250;
		if (yy>-midv&&yy<=midv) return 1;
		else return 0;
	}
	if (~f[x][y][t]) return g[x][y][t];
	LL ans=0,num=0;
	if (x==mid) {
		if (!t||midv<=a[x]) {
			ans+=Solve(x+1,y,t&(midv==a[x]));
			num+=dp(x+1,y,t&(midv==a[x]));
		}
	}
	else if (t) {
		for (int i=0;i<=a[x];i++) {
			int sum=i*((x>mid)?-1:1);
			LL t=Solve(x+1,y+sum,i==a[x]);
			ans+=t;
			num+=abs(x-mid)*i*t+dp(x+1,y+sum,i==a[x]);
		}
	}
	else if (!t) {
		for (int i=0;i<base;i++) {
			int sum=i*((x>mid)?-1:1);
			LL t=Solve(x+1,y+sum,0);
			ans+=t;
			num+=abs(x-mid)*i*t+dp(x+1,y+sum,0);
		}
	}
	f[x][y][t]=num;
	return g[x][y][t]=ans;
}

LL calc(LL x) {
	cnt=0;
	for (;x;x/=base) a[++cnt]=x%base;
	for (int i=1;i*2<=cnt;i++) swap(a[i],a[cnt-i+1]);
	LL ans=0;
	for (mid=1;mid<=cnt;mid++) {
		for (midv=1;midv<base;midv++) {
			memset(f,0xff,sizeof(f));
			memset(g,0,sizeof(g));
			Solve(1,250,1);
			ans+=f[1][250][1];
		}
	}
	return ans;
}

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