前缀和与差分——海底高铁

前缀和与差分——海底高铁

题目背景

大东亚海底隧道连接着厦门、新北、博艾、那霸、鹿儿岛等城市，横穿东海，耗资 1000 亿博艾元，历时 15 年，于公元 2058 年建成。凭借该隧道，从厦门可以乘坐火车直达台湾、博艾和日本，全程只需要 4 个小时。

题目描述

该铁路经过 N 个城市，每个城市都有一个站。不过，由于各个城市之间不能协调好，于是乘车每经过两个相邻的城市之间（方向不限），必须单独购买这一小段的车票。第i段铁路连接了城市i和城市 i+1(1 <= i < N)。如果搭乘的比较远，需要购买多张车票。第 i 段铁路购买纸质单程票需要 Ai 博艾元。

虽然一些事情没有协调好，各段铁路公司也为了方便乘客，推出了 IC 卡。对于第 i 段铁路，需要花 Ci 博艾元的工本费购买一张 IC 卡，然后乘坐这段铁路一次就只要扣 Bi(Bi < Ai)元。IC 卡可以提前购买，有钱就可以从网上买得到，而不需要亲自去对应的城市购买。工本费不能退，也不能购买车票。每张卡都可以充值任意数额。对于第 i 段铁路的 IC 卡，无法乘坐别的铁路的车。

Uim 现在需要出差，要去 M 个城市，从城市 P1 出发分别按照 P1,P2,P3…PM 的顺序访问各个城市，可能会多次访问一个城市，且相邻访问的城市位置不一定相邻，而且不会是同一个城市。

现在他希望知道，出差结束后，至少会花掉多少的钱，包括购买纸质车票、买卡和充值的总费用。

输入格式

第一行两个整数，N，M。

接下来一行，M个数字，表示Pi

接下来 N - 1 行，表示第i段铁路的 Ai,Bi,Ci

输出格式

一个整数，表示最少花费

输入输出样例

输入样例

9 10
3 1 4 1 5 9 2 6 5 3
200 100 50
300 299 100
500 200 500
345 234 123
100 50 100
600 100 1
450 400 80
2 1 10

输出样例

6394

说明/提示

2 到 3 以及 8 到 9 买票，其余买卡。

对于 30% 数据 M = 2

对于另外 30% 数据 N <= 1000 ，M <= 1000

对于100%的数据 M,N <= 100000,Ai,Bi,Ci <= 100000

分析

可以先用前缀和与差分求出每一段需要经过的次数，然后再用贪心策略在两种买票方式上进行选取。

#include<bits/stdc++.h>
using namespace std;
long long n,m,p[100005],a[100005],b[100005],c[100005],d[100005];

int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		cin>>p[i];
	}
	for(int i=1;i<=n-1;i++){
		cin>>a[i]>>b[i]>>c[i];
	}
	memset(d,0,sizeof(d));
	for(int i=1;i<m;i++){
		if(p[i+1]<p[i]){
			d[p[i+1]]++;
			d[p[i]]--;
		}
		else{
			d[p[i]]++;
			d[p[i+1]]--;
		}
	}
	for(int i=1;i<=n;i++){
		d[i]+=d[i-1];
	}
	long long sum=0;
	for(int i=1;i<=n-1;i++){
		sum+=min(a[i]*d[i],b[i]*d[i]+c[i]);
	}
	cout<<sum<<endl;
	return 0;
}