题目大意:
在一维坐标轴上,有
V
V
V个村庄,在坐标轴上选择放置
P
P
P个邮局,计算每个村庄和最近的邮局之间所有距离的最小可能总和。
1
≤
P
≤
300
,
P
≤
V
≤
3000
,
1
≤
村
庄
位
置
≤
10000
1 \le P \le 300, P \le V \le 3000, 1 \le 村庄位置 \le 10000
1≤P≤300,P≤V≤3000,1≤村庄位置≤10000
解题思路:
- 设 d p [ i ] [ j ] dp[i][j] dp[i][j]表示前 i i i个村庄放 j j j个邮局的最小距离总和, w ( i , j ) w(i,j) w(i,j)表示村庄区间 [ i , j ] [i,j] [i,j]内放一个村庄时该区间的最小距离总和。
- d p [ i ] [ j ] = m i n { d p [ k ] [ j − 1 ] + w ( k + 1 , i ) } , k ∈ [ 0 , i ) dp[i][j] = min\{dp[k][j-1] + w(k+1,i)\}, k \in [0, i) dp[i][j]=min{dp[k][j−1]+w(k+1,i)},k∈[0,i)
- 而村庄数为奇数,放中位数距离和最小,村庄数为偶数,放中间两个村庄之间任意一处均可。
- w ( i , j ) = w ( i , j − 1 ) + x [ j ] − x [ ⌊ i + j 2 ⌋ ] w(i,j)=w(i,j-1)+x[j]-x[\left \lfloor \frac {i+j}{2} \right \rfloor] w(i,j)=w(i,j−1)+x[j]−x[⌊2i+j⌋]
- 设 d ( i , j ) d(i,j) d(i,j)为 d p ( i , j ) dp(i,j) dp(i,j)取得最优值时对应的下标
- 通过暴力打表可以发现 w ( i , j ) w(i,j) w(i,j)具有四边形不等式,且满足区间包含单调性,那么dp也满足四边形不等式
- 所以 d ( i , j ) ≤ d ( i , j + 1 ) ≤ d ( i + 1 , j + 1 ) d(i,j)\le d(i,j+1)\le d(i+1,j+1) d(i,j)≤d(i,j+1)≤d(i+1,j+1)
AC代码:
#include <bits/stdc++.h>
#define ft first
#define sd second
#define IOS ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define seteps(N) fixed << setprecision(N)
#define endl "\n"
const int maxv = 3e3 + 10;
const int maxp = 3e2 + 10;
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int, int> pii;
const ll mod = 1e9 + 7;
const ll inf = 0x3f3f3f3f3f;
ll qpow(ll a, ll b) {
ll res = 1;
while (b) {
if (b & 1) res = res * a % mod;
a = a * a % mod;
b >>= 1;
}
return res;
}
int v, p, x[maxv], d[maxv][maxp];
ll w[maxv][maxv], dp[maxv][maxp];
int main() {
scanf("%d%d", &v, &p);
for (int i = 1; i <= v; i++) scanf("%d", &x[i]);
sort (x + 1, x + v + 1);
for (int l = 2; l <= v; l++)
for (int i = 1, j; i + l - 1 <= v; i++) {
j = i + l - 1;
w[i][j] = w[i][j - 1] + x[j] - x[(i + j) / 2];
}
memset(dp, 20, sizeof(dp));
dp[0][0] = 0;
for (int j = 1; j <= p; j++) {
d[v + 1][j] = v;
for (int i = v; i >= 1; i--) {
ll mmin = inf; int minid;
for (int k = d[i][j - 1]; k <= d[i + 1][j]; k++) {
if (dp[k][j - 1] + w[k + 1][i] < mmin)
mmin = dp[k][j - 1] + w[k + 1][i], minid = k;
}
dp[i][j] = mmin;
d[i][j] = minid;
}
}
cout << dp[v][p] << endl;
}