72分求助大佬
查看原帖
72分求助大佬
194200
gcs1237楼主2020/5/10 11:19
#include<bits/stdc++.h>
using namespace std;
long long n,m,a[200001],s,ss,p,q,l,h,ans,mi=10000000000;
int main() {
	cin>>n;
	for(int i=1; i<=n; i++)
		cin>>a[i];
	cin>>m>>p>>s>>ss;
	a[p]+=s;
	for(int i=1; i<m; i++)
		l+=(m-i)*a[i];
	for(int i=m+1; i<=n; i++)
		h+=(i-m)*a[i];
	if(l==h&&ss!=0) {
		cout<<m;
		return 0;
	}
	if(h>l) {
		for(int i=1; i<m; i++) {
			q=l+ss*(m-i);
			if(abs(q-h)<mi) {
				ans=i;
				mi=abs(q-h);
			}
		}
	} else {
		for(int i=m+1; i<=n; i++) {
			q=h+ss*(i-m);
			if(abs(q-l)<mi) {
				mi=abs(q-l);
				ans=i;
			}
		}
	}
	cout<<ans;
	return 0;
}

啊啊啊!!!

2020/5/10 11:19
加载中...