题目有点绕,注意读题的时候提取关键信息。
分析:既然是要按编号顺序进行,就有了一个固定的顺序,可以 DP。而且要注意:
由上图可以发现,对于每个点
i
,都能找到一个点
定义
很显然 d(i,i+1) 可以预先计算,因此上式中两个连续求和都可以用前缀和做到 O(1),不妨令
对于上式直接枚举
i
和
参考代码:
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
using namespace std;
const int MAXN = 5e4 + 100;
int N, C;
int X[MAXN], Y[MAXN], Q[MAXN], sumq[MAXN];
double d[MAXN], sumd[MAXN], f[MAXN];
inline double calc(int i, int j) {
return hypot(X[i] - X[j], Y[i] - Y[j]);
}
int main(void) {
freopen("1763.in", "r", stdin);
freopen("1763.out", "w", stdout);
scanf("%d%d", &N, &C);
X[0] = Y[0] = 0; sumq[0] = 0;
for (int i = 1; i <= N; i++) {
scanf("%d%d%d", &X[i], &Y[i], &Q[i]);
sumq[i] = sumq[i - 1] + Q[i];
d[i] = calc(i - 1, i);
sumd[i] = sumd[i - 1] + d[i];
}
for (int i = 1; i <= N; i++) {
f[i] = f[i - 1] + calc(0, i) * 2;
for (int j = i - 2; j >= 0 && sumq[i] <= sumq[j] + C; j--)
f[i] = min(f[i], f[j] + calc(0, j + 1) + (sumd[i] - sumd[j + 1]) + calc(i, 0));
}
printf("%.2lf\n", f[N]);
return 0;
}
下面来考虑通过优化这个 DP 获得正解。
不难发现,可以将上式中涉及 i 的部分单独提出来,则变形为
很显然
sumqj
是单调递增的,由后面的限制条件可知,计算
fi
时决策
j
的起点是单调不递减的,因此本质上是维护一个限定
参考代码:
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <deque>
using namespace std;
typedef pair <int, double> pid;
const int MAXN = 5e4 + 100;
int N, C;
int X[MAXN], Y[MAXN], Q[MAXN], sumq[MAXN];
double d[MAXN], sumd[MAXN], f[MAXN];
deque <pid> dq;
inline double calc(int i, int j) {
return hypot(X[i] - X[j], Y[i] - Y[j]); //注意:这个函数不是 ANSI C 的
}
void print_queue() {
deque<pid>::iterator it = dq.begin();
while (it != dq.end()) {
printf("(%d,%.2lf)", it->first, it->second);
it++;
}
putchar('\n');
}
int main(void) {
freopen("1763.in", "r", stdin);
freopen("1763.out", "w", stdout);
scanf("%d%d", &N, &C);
X[0] = Y[0] = 0; sumq[0] = sumd[0] = 0;
for (int i = 1; i <= N; i++) {
scanf("%d%d%d", &X[i], &Y[i], &Q[i]);
sumq[i] = sumq[i - 1] + Q[i];
sumd[i] = sumd[i - 1] + calc(i - 1, i);
// printf("to O:%.2lf sumd:%.2lf to last:%.2lf\n", calc(0, i), sumd[i], sumd[i] - sumd[i - 1]);
}
dq.push_back(make_pair(0, 0.0));
for (int i = 1; i <= N; i++) {
while (!dq.empty() && sumq[i] > sumq[dq.front().first] + C) dq.pop_front();
// printf("i=%d ", i);print_queue();
f[i] = sumd[i] + calc(i, 0) + dq.front().second;
// printf("%.2lf\n", f[i]);
double t = f[i] + calc(0, i + 1) - sumd[i + 1]; //要利用单调队列维护的部分,注意下标不能都写成 i,要跟原来的式子一致
// printf("back=%.2lf\n", dq.back().second);
while (!dq.empty() && t < dq.back().second) dq.pop_back();
dq.push_back(make_pair(i, t));
// print_queue();
}
printf("%.2lf\n", f[N]);
return 0;
}
后记:这题我从 60 分改到 100 分的时候,中间爆零了两次,最后发现我维护队尾的时候把 while (!dq.empty() && t < dq.back().second) 里面的 back 打成了 front……调了半天之后发现这个傻*错误,什么也不想说了。