10分求调,按题解的意思来的,难道我理解错了吗?
查看原帖
10分求调,按题解的意思来的,难道我理解错了吗?
213192
niubi_aaa楼主2024/6/17 17:26
#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[200001],pre[200001],nex[200001];
long long ans=0;
struct node{
	int k,x,y;
}nb[200001];
bool cmp(node a,node b){
	return a.k>b.k; 
}
int main(){
	scanf("%d%d",&n,&m);
	memset(pre,0,sizeof(pre));
	memset(nex,0,sizeof(nex));
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<=m;i++)scanf("%d%d%d",&nb[i].x,&nb[i].y,&nb[i].k);
	sort(nb+1,nb+1+m,cmp);
	for(int i=1;i<=n;i++){
		if(i==1){
			pre[i]=i;
			continue;
		}
		if(a[i]<=a[pre[i-1]])pre[i]=i;
		else pre[i]=pre[i-1];
	}
	for(int i=n;i>=1;i--){
		if(i==n){
			nex[i]=i;
			continue;
		}
		if(a[i]<=a[nex[i+1]])nex[i]=i;
		else nex[i]=nex[i+1];
	}
	int maxx=nb[1].x,miny=nb[1].y;
	int zz[200001],pd1=0;
	memset(zz,0,sizeof(zz));
	for(int i=2;i<=m;i++){
		if(nb[i].k!=nb[i-1].k)zz[++pd1]=nb[i-1].k-nb[i].k;
	}
	if(m==1)ans=ans+a[nex[maxx]]*nb[1].k+a[pre[miny]]*(nb[1].k-1);
	int pd=nb[1].k,pd2=1;
	for(int i=2;i<=m;i++){
		if(nb[i].k==pd){
			maxx=max(maxx,nb[i].x);
			miny=min(miny,nb[i].y);
		}
		else{
			ans=ans+a[nex[maxx]]*zz[pd2]+a[pre[miny]]*zz[pd2];
			pd2++;
			pd=nb[i].k;
			maxx=max(maxx,nb[i].x);
			miny=min(miny,nb[i].y);
		}
		if(i==m){
			ans=ans+a[nex[maxx]]*nb[i].k+a[pre[miny]]*(nb[i].k-1);
		}
	}
	printf("%lld",ans);
	return 0;
} 
2024/6/17 17:26
加载中...