80pts求助
查看原帖
80pts求助
1143897
xingjielong楼主2024/10/18 20:33

先计算不做调整的势力,再打擂台求最值

#include<bits/stdc++.h>
using namespace std;
long long a[100100];
long long m,n,p1,s1,s2;
long long shili(long long i){//计算点i的势力 
	return abs(i-m)*a[i];
}
int main(){
	cin >> n;
	for(long long i=1;i<=n;i++)cin >> a[i];
	cin >> m >> p1 >> s1 >> s2;
	int lsum=0,rsum=0;//天降神兵后的龙实力天降神兵的虎实力 
	a[p1]+=s1;//天降神兵 
	for(long long i=1;i<m;i++){
		lsum+=shili(i);
	}
	for(long long i=m+1;i<=n;i++){
		rsum+=shili(i);
	}
	long long ans=m;//默认为m位置 
	long long ansl=lsum;//ans情况的龙实力 
	long long ansr=rsum;//ans情况的虎实力 
	for(long long i=1;i<m;i++){//枚举龙的地盘 
		lsum+=abs(i-m)*s2;
		if(abs(lsum-rsum)<abs(ansl-ansr)){
			ansl=lsum;
			ansr=rsum;
			ans=i;
		}
		lsum-=abs(i-m)*s2;
	}
	for(long long i=m+1;i<=n;i++){//枚举虎的地盘 
		rsum+=abs(i-m)*s2;
		if(abs(lsum-rsum)<abs(ansl-ansr)){
			ansl=lsum;
			ansr=rsum;
			ans=i;
		}
		rsum-=abs(i-m)*s2;
	}
	cout << ans;
	return 0;
}
2024/10/18 20:33
加载中...