蒟蒻求助【QaQ】 P5016

#include<bits/stdc++.h>
using namespace std;
int n,m,p1,s1,s2,a[100005];
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	cin>>m>>p1>>s1>>s2;
	int ds=0,ts=0;
	a[p1]+=s1;
	int p2=0,minn=100005;
	for(int i=1;i<=n;i++){
		if(i<m) ds+=(m-i)*a[i];
		else if(i>m) ts+=(i-m)*a[i];
	}
	for(int i=1;i<=n;i++){
		if(i<m) ds+=(m-i)*s2;
		else if(i>m) ts+=(i-m)*s2;
		if(abs(ds-ts)<minn) minn=abs(ds-ts),p2=i;
		if(i<m) ds-=s2*(m-i);
		else if(i>m) ts-=s2*(i-m);
	}
	cout<<p2<<endl;
	return 0;
}