题解：P8632 [蓝桥杯 2015 国 B] 居民集会

洛谷阅读更佳

https://www.luogu.com.cn/article/32sdhywg

题意

给定 $n$ 个家庭，其中每个家庭距离起点有 $d_i$ 的距离，每个家庭有 $t_i$ 个人。

题目需要我们在长度为 $L$ 的数轴上放置四个点 $p_1,p_2,p_3,p_4(p_1\le p_2\le p_3\le p_4)$ 并且最后一个点必须落在 $L$ 上，然后这 $n$ 个家庭会往右一直走直到碰到任意一个点，设第 $i$ 个家庭到达一个点的距离是 $le{n}_i$ 那么这个家庭的开销是 $le{n}_i\ast t_i$ 。

问如何安排这些点使得所有家庭的开销总和最小？

思路

首先贪心地想，这些点肯定是落在某个家庭的位置代价最小。

然后来考虑动态规划，设 $dp[i][j]$ 为以第 $i$ 个家庭结尾，并且放置了 $j$ 个点的最小开销，那么转移方程就是从前面的数中找到一个 $k$ 作为 $j-1$ 的结尾，然后 $dp[i][j]=dp[k][j-1]+[k+1,i]的开销$ 。

那么问题来了，如何计算一个区间的开销呢？

设 $pre{s}_i$ 为只放置一个点在 $i$ 并且以 $i$ 结尾时所有家庭的开销。假设我已经知道 $pre{s}_{i-1}$ 了，那么我如何计算出 $pre{s}_i$ 呢？经过观察，对于 $pre{s}_{i-1}$ 来说，每多走一个单位距离，就会使得前面所有的家庭又多走了一个单位距离也就是增加了一倍的家庭人数前缀和。所以设 $T_i={\sum }_{j=1}^i{t}_j$ ，得：
$$pre{s}_i=pre{s}_{i-1}+T_{i-1}\ast (d_i-d_{i-1})$$

得到所有的 $pre{s}_i$ 后，获取一个区间的开销就变轻松了：

$$pre{s}_{[l,r]}=pre{s}_r-pre{s}_{l-1}-(d_r-d_{l-1})\ast T_{l-1}$$

也就是把 $l$ 之前的贡献减掉就好了。

回到我们的状态转移方程，有：

$$dp[i][j]=\underset{k=0}{\overset{i-1}{\min }}(dp[k][j-1]+pre{s}_i-pre{s}_k-(d_i-d_k)\ast T_k)$$

这样的复杂度是 $n^2$ 显然是会T的，不要着急，我们思考一下点开标签，考虑动态规划优化。整理转移方程的式子得（过程读者可自行推导）：

$$dp[k][j-1]-pre{s}_k+T_k\ast d_k=T_k\ast d_i-pre{s}_i+dp[i][j]$$

是不是有点眼熟？有点像形如 $y=kx+b$ 的式子？我们令：

$y=dp[k][j-1]-pre{s}_k+T_k\ast d_k,$

$x=T_k,$

$k=d_i,$

$b=dp[i][j]-pre{s}_i$

那么对于这个式子来说，只要使得 $b$ 最小，那么 $dp[i][j]$ 就最小，画一个二维坐标系发现横坐标单调递增，所以可以使用优先队列维护我们的答案。

注意的点

运算过程中会爆LL所以使用int128防溢出。

代码

#include <iostream>
using namespace std;
#define MAX_N 100005
#define ll long long
#define ull unsigned ll
ll dp[5][MAX_N] = { 0 };
ll dis[MAX_N] = { 0 };
__int128 T[MAX_N] = { 0 };
__int128 pres[MAX_N] = { 0 };
int head = 1, tail = 1;
int q[MAX_N + 10] = { 0 };
__int128 get_y(int i, int j) { // j - 1
	return dp[j - 1][i] - pres[i] + T[i] * dis[i];
}
long double slope(int a, int b, int j) {
	ll x1 = T[a], x2 = T[b];
	ll y1 = get_y(a, j), y2 = get_y(b, j);
	if (x1 == x2) return 999999999999999;
	return (y2 - y1) * 1.0f / (x2 - x1);
}
int main() {
	int n, L;
	scanf("%d%d", &n, &L);
	for (int x = 1; x <= n; x++) {
		ll ti;
		scanf("%lld%lld", dis + x, &ti);
		T[x] = ti + T[x - 1];
		pres[x] = pres[x - 1] + T[x - 1] * (dis[x] - dis[x - 1]);
		//cout << pres[x] << endl;
	}
	n++;
	dis[n] = L;
	T[n] = T[n - 1];
	pres[n] = pres[n - 1] + T[n - 1] * (dis[n] - dis[n - 1]);
	for (int i = 1; i <= n; i++) dp[1][i] = pres[i];
	for (int j = 2; j <= 4; j++) {
		head = tail = 1;
		q[tail++] = 0;
		for (int i = 1; i <= n; i++) {
			ll xielv = dis[i];
			while (tail - head >= 2 && slope(q[head], q[head + 1], j) < 
			xielv)
				head++;
			//printf("[%d, %d] head = %d\n", i, j, q[head]);
			int k = q[head];
			dp[j][i] = dp[j - 1][k] + pres[i] - pres[k] - (dis[i] - dis[k]) * T[k];
			//printf("%llu   %llu   %llu\n", dis[i], dis[k], T[k]);
			while (tail - head >= 2 && slope(q[tail - 1], q[tail - 2], j) >=
				slope(q[tail - 1], i, j))
				tail--;
			q[tail++] = i;
			//printf("[%d, %d] = %lld\n", i, j, dp[j][i]);
		}
	}
	ll ans = 0x7f7f7f7f7f7f7f7f;
	for (int x = 1; x <= 4; x++) ans = min(ans, dp[x][n]);
	printf("%lld", ans);
	return 0;
}