算法
(贪心,递推) $O(nk)$
这道题目的信息较多,我们先将其整理一下。
首先预处理出每个站台的最早发车时间last[i]
,即最后一个到达站台i
的时间。然后预处理出从每个站台下车的人数sum[i]
。
接下来求出车最早到达每个站台的时间tm[i]
,tm[i] = max(tm[i - 1], last[i - 1]) + d[i - 1]
, 其中 d[i - 1]
是从第i - 1
个站台走到第i
个站台的时间。
那么每个乘客的旅行时间就是 tm[b[i]] - t[i]
,其中b[i]
是乘客的终点站,t[i]
是乘客到达起点的时间。
然后考虑每个氮气加速器该用在哪一段,这里使用贪心策略:每次选择当前节约时间最多的一段即可。下面给出证明:
我们将所有加速之后可以提速的连续景点用红色标注,可以得到下图。注意这里每段红色线段不是指相邻两个景点之间的路程,而是指一段连续的可以提速的景点。
从中可以发现如下几个性质:
- 每次加速一段之后,可能会影响接下来一段连续的站点。即在每段红色区间中的任意一个站点加速,都可以将该点及右边剩余红色部分的站点全部加速。因此在区间内部,加速最左端的站点一定是最优的。
- 不同红色区间之前完全独立,加速其中某个区间时,对其余区间没有任何影响。
- 加速某个区间左端点之后,该区间可能会分裂成两个子区间,这两个子区间的加速效果小于等于原区间的加速效果。
因此,当前加速效果最好的一段一定可以出现在最优解中,否则可以将最优解中的某个区间替换成加速效果最好的区间,答案不会变差。
时间复杂度
算法瓶颈在氮气加速这部分。一共有 $k$ 个氮气加速器,每次需要递推出在每一段加速后降低的总时间,这一步需要 $O(n)$ 的计算量。
因此总时间复杂度是 $O(nk)$。
C++ 代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010, M = 10010;
int n, m, k;
int d[N];
int t[M], a[M], b[M];
int last[N], sum[N];
int tm[N], reduce[N];
int main()
{
scanf("%d%d%d", &n, &m, &k);
for (int i = 1; i < n; i++) scanf("%d", &d[i]);
for (int i = 0; i < m; i++)
{
scanf("%d%d%d", &t[i], &a[i], &b[i]);
last[a[i]] = max(last[a[i]], t[i]);
sum[b[i]] ++;
}
for (int i = 1; i <= n; i++) tm[i] = max(tm[i - 1], last[i - 1]) + d[i - 1];
while (k--)
{
for (int i = n; i >= 2; i--)
{
reduce[i - 1] = sum[i];
if (tm[i] > last[i]) reduce[i - 1] += reduce[i];
}
int p = 0;
for (int i = 1; i <= n; i++)
if (d[i] && reduce[p] < reduce[i])
p = i;
d[p] --;
for (int i = p; i <= n; i++) tm[i] = max(tm[i - 1], last[i - 1]) + d[i - 1];
}
int res = 0;
for (int i = 0; i < m; i++) res += tm[b[i]] - t[i];
printf("%d\n", res);
return 0;
}
自己按照这个思路写了一遍,被落谷的最后一个$Hack$数据卡掉了。
把y总的代码直接交了一遍,结果还是被$Hack$了。
我下载了$Hack$数据,是这样的:
$Input$:
$Output$:
$My Output$:
然后经过调试,我发现程序选择了$d[3]$这个地方加氮气加速器。
原因是因为$d[2] = 0$,故$4$的需求没有传递到$2$。所以即使所需时间为$0$,只要满足条件,前后两段也是可以串联的。只不过不能在这个地方使用加速器,找最大值时,忽略这些为$0$的即可。
代码改动如下(31 ~ 41行):
代码已改hh
感谢指正!
y大辛苦啦~另我已经报名算法基础课好久啦,想进群,求邀请码QAQ
已私信~