海底高铁 前缀和+贪心

N N N个城市,每个城市之间有一段铁路,经过两个相邻的城市之间,必须单独购买这一小段的车票,第 i i i段铁路购买纸质单程票需要 A i A_i Ai元,或购买卡,对于第 i i i段铁路,需要花 C i C_i Ci元的工本费购买一张卡,然后乘坐这段铁路一次就只要扣 B i ( B i < A i ) ​ B_i(B_iBi(Bi<Ai)元。现要去 M M M个城市,求最少会花掉多少的钱。
对于每一小段铁路要么全部买纸票,要么全部刷卡。我们需要统计第 i i i段铁路要经过多少次,再比较两者谁便宜。本题数据比较大,需要使用前缀和来统计区间和。

#include
using namespace std;
long long a[100005],b[100005],c[100005],num[100005];
long long ans[100005];
long long sum;
int n,m;
int main()
{
	cin>>n>>m;
	num[0]=0;
	for(int i=1;i<=m;i++)cin>>num[i];
	for(int i=1;i<=n-1;i++)cin>>a[i]>>b[i]>>c[i];
	for(int i=1;i<=m-1;i++)
	{
		int N,M;
		M=max(num[i],num[i+1]);N=min(num[i],num[i+1]);
		ans[N]++;ans[M]--;
	}	
	for(int i=1;i<=n;i++)ans[i]+=ans[i-1];
//	for(int i=1;i<=n;i++)cout<
	for(int i=1;i<=n-1;i++)
	sum+=min(a[i]*ans[i],(b[i]*ans[i]+c[i]));
	cout<<sum<<endl;
	cin>>n;
	return 0;
}

你可能感兴趣的:(算法,c++,数据结构)