52分QWQ求救SOS

#include <bits/stdc++.h>
using namespace std;
long long c[100005],m;
int n,p1,s1,s2;
int l=0,h=0;
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>c[i];
	}
	cin>>m>>p1>>s1>>s2;
	c[p1]+=s1;
	for(int i=1;i<m;i++){//求龙方气势 
		l+=c[i]*(m-i);
	}
	for(int i=m+1;i<=n;i++){//求虎方气势
		h+=c[i]*(i-m); 
	}
	long long zy=abs(l-h);
	//cout<<l<<" "<<h<<" ";//确定龙<虎 
	int p;
	if(l<h){//龙<虎 
		for(int i=m-1;i>0;i--){
			if(abs(h-l-(m-i)*s2)<=zy){
				p=i;
				zy=abs(h-l-(m-i)*s2);
			}
		}
	}else if(l>h){//龙>虎
		for(int i=n;i>m;i--){
			if(abs(l-h-(i-m)*s2)<=zy){
				p=i;
				zy=abs(l-h-(i-m)*s2);
			}
		}
	}else{//龙=虎 
		p=(s2==0?1:m);
	}
	cout<<p;
	return 0;
}