P3406 海底高铁（差分）

题目描述

分析

差分求出每段路程的经过次数，在根据a,b,c求出哪种方案便宜

for(int i = 1;i<m;i++){
		int x= min(a[i],a[i+1]),y = max(a[i],a[i+1]);
		b[  x  ] += 1;
		b[  y  ] -= 1;	
	} 

ps. 因为是对路程差分，所以是b[y]-- ,不是b[y+1]–。

代码

#include<iostream>
#include<stdlib.h>
#include<string>
#include<algorithm>
#include<vector>
#include<cstdio>
using namespace std;
const int maxn = 1e5+50;
long long n,m,a[maxn],b[maxn],t1,t2,t3;
long long ans;
int main(){
	//freopen("a.txt","r",stdin);
	cin>>n>>m;
	for(int i =1;i<=m;i++){
		cin>>a[i];
	}
	
	//差分标记 
	for(int i = 1;i<m;i++){
		int x= min(a[i],a[i+1]),y = max(a[i],a[i+1]);
		b[  x  ] += 1;
		b[  y  ] -= 1;	
	} 
	
	//前缀和,得出每段路经过的次数 
	for(int i = 1;i<=n-1;i++){
		b[i] += b[i-1];
	} 
	
	for(int i = 1;i<=n-1;i++){
		cin>>t1>>t2>>t3;
		long long pa = t1*b[i],pb = t3 + t2*b[i];
		ans += min(pa,pb);
		//cout<<ans<<endl;
	} 
	cout<<ans<<endl;
	return 0;
}