【洛古 P1315】 [NOIP2011 提高组] 观光公交

[NOIP2011 提高组] 观光公交

题目背景

感谢 @Transhumanist 提供的一组 Hack 数据

题目描述

风景迷人的小城 Y 市，拥有 $n$ 个美丽的景点。由于慕名而来的游客越来越多，Y 市特意安排了一辆观光公交车，为游客提供更便捷的交通服务。观光公交车在第 $0$ 分钟出现在 $1$ 号景点，随后依次前往 $2,3,4,\cdots ,n$ 号景点。从第 $i$ 号景点开到第 $i+1$ 号景点需要 $D_i$ 分钟。任意时刻，公交车只能往前开，或在景点处等待。

设共有 $m$ 个游客，每位游客需要乘车 $1$ 次从一个景点到达另一个景点，第 $i$ 位游客在 $T_i$ 分钟来到景点 $A_i$，希望乘车前往景点 $B_i$（$A_i<B_i$）。为了使所有乘客都能顺利到达目的地，公交车在每站都必须等待需要从该景点出发的所有乘客都上车后才能出发开往下一景点。

假设乘客上下车不需要时间。一个乘客的旅行时间，等于他到达目的地的时刻减去他来到出发地的时刻。因为只有一辆观光车，有时候还要停下来等其他乘客，乘客们纷纷抱怨旅行时间太长了。于是聪明的司机 ZZ 给公交车安装了 $k$ 个氮气加速器，每使用一个加速器，可以使其中一个 $D_i-1$。对于同一个 $D_i$ 可以重复使用加速器，但是必须保证使用后 $D_i\ge 0$。

那么 ZZ 该如何安排使用加速器，才能使所有乘客的旅行时间总和最小？

输入格式

第 $1$ 行是 $3$ 个整数 $n,m,k$，每两个整数之间用一个空格隔开。分别表示景点数、乘客数和氮气加速器个数。

第 $2$ 行是 $n-1$ 个整数，每两个整数之间用一个空格隔开，第 $i$ 个数表示从第 $i$ 个景点开往第 $i+1$ 个景点所需要的时间，即 $D_i$。

第 $3$ 行至 $m+2$ 行每行 $3$ 个整数 $T_i,A_i,B_i$，每两个整数之间用一个空格隔开。第 $i+2$ 行表示第 $i$ 位乘客来到出发景点的时刻，出发的景点编号和到达的景点编号。

输出格式

一个整数，表示最小的总旅行时间。

样例 #1

样例输入 #1

3 3 2
1 4
0 1 3
1 1 2
5 2 3

样例输出 #1

10

提示

【输入输出样例说明】

对 $D_2$ 使用 $2$ 个加速器，从 $2$ 号景点到 $3$ 号景点时间变为 $2$ 分钟。

公交车在第 $1$ 分钟从 $1$ 号景点出发，第 $2$ 分钟到达 $2$ 号景点，第 $5$ 分钟从 $2$ 号景点出发，第 $7$ 分钟到达 $3$ 号景点。

第 $1$ 个旅客旅行时间 $7-0=7$ 分钟。

第 $2$ 个旅客旅行时间 $2-1=1$ 分钟。

第 $3$ 个旅客旅行时间 $7-5=2$ 分钟。

总时间 $7+1+2=10$ 分钟。

【数据范围】

对于 $10\%$ 的数据，$k=0$。

对于 $20\%$ 的数据，$k=1$。

对于 $40\%$ 的数据，$2\le n\le 50$，$1\le m\le 10^3$，$0\le k\le 20$，$0\le D_i\le 10$，$0\le T_i\le 500$。

对于 $60\%$ 的数据，$1\le n\le 100$，$1\le m\le 10^3$，$0\le k\le 100$，$0\le D_i\le 100$，$0\le T_i\le 10^4$。

对于 $100\%$ 的数据，$1\le n\le 10^3$，$1\le m\le 10^4$，$0\le k\le 10^5$，$0\le D_i\le 100$，$0\le T_i\le 10^5$。

---

解题思路

首先我们来考虑$k=0$的情况，这时候t[i]为到第i个景点上车的乘客最晚的时间，last[i]为到车到达第i个景点的最小时间。

t[a[i]]=max(t[a[i]],T[i]); //T[i]为第i个乘客到达景点等车的时刻。

得到t可地推出last： last[i]=max(last[i-1],t[i-1])+d[i-1];

那么答案即为：${\sum }_{i=1}^{i=m}last[b[i]]-T[i]$
这时题目说可以缩短公交车的时间并且最多缩短K次，相当于可以贪心K次。
我们考虑缩短哪一段距离后，使得尽可能多的乘客旅行时间减一

如果我们在 $i$ 到$i+1$间使用加速器的话，那么在$i+1$站下车的乘客的旅行时间都会减一，但是如果$last[i+1]$小于$t[i+1]$的话，车就要在$i+1$站等到所有的乘客上车，在$i+1$站以后下车的乘客的时间是一样的，这个加速器对后面下车的乘客没有影响。我们可以用递推求出每个$i$最远能影响的车站，用$f[i]$表示，预处理求出在前i个车站中下车的总人数，用$sum[i]$表示

if(last[i+1]>t[i+1])
	f[i]=f[i+1];
else 	f[i]=i+1;

枚举每个车站，取$sum[f[i]]-sum[i]$的最大值，就是这次贪心减小的时间，对应试用加速器的道路时间-1.（注意道路时间不能为负）

每次贪心玩要重新计算一遍last，因为道路时间改变了。

---

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;

int n,m,k,ans;
int a[10010],b[10010],T[10010],t[1010],f[1010],d[1010],sum[1010],last[1010];

int main(){
	freopen("bus.in","r",stdin);
	freopen("bus.out","w",stdout);
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<n;i++)
	{
		scanf("%d",&d[i]);
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&T[i],&a[i],&b[i]);
		ans=ans-T[i];
		sum[b[i]]++;
		t[a[i]]=max(t[a[i]],T[i]);
	}
	for(int i=2;i<=n;i++)
	{
		last[i]=max(last[i-1],t[i-1])+d[i-1];
		sum[i]+=sum[i-1];
	}
	for(int i=1;i<=m;i++)
		ans=ans+last[b[i]];
	while(k--)
	{
		f[n-1]=f[n]=n;
		for(int i=n-2;i>0;i--)
		{
			if(last[i+1]>t[i+1])f[i]=f[i+1];
			else f[i]=i+1;
		}
		int id=0,maxn=-1;
		bool ok=1;
		for(int i=1;i<n;i++)
		{
			if(maxn<sum[f[i]]-sum[i]&&d[i])
			{
				maxn=sum[f[i]]-sum[i],id=i;
				ok=0;
			}
		}
		if(ok)break;
		ans=ans-maxn,d[id]--;
		for(int i=2;i<=n;i++)
			last[i]=max(last[i-1],t[i-1])+d[i-1];
	}
	printf("%d",ans);
}
/*
3 3 2
1 4
0 1 3
1 1 2
5 2 3

*/