最后五个点WA了,哪位大佬帮我看一下
#include<bits/stdc++.h>
using namespace std;
long long a[9999999],n,m,p1,s1,s2,b1,b2,ans,num;
int main(){
scanf("%lld",&n);
for(int i=1;i<=n;++i)
scanf("%lld",&a[i]);
scanf("%lld%lld%lld%lld",&m,&p1,&s1,&s2);
a[p1]+=s1;
for(int i=1;i<m;++i)
b1+=a[i]*(m-i);
for(int i=m+1;i<=n;++i)
b2+=a[i]*(i-m);
ans=m;
if(b1>b2){
int h;
num=b1-b2;
for(int i=m+1;i<=n;++i){
h=abs(b1-b2-(i-m)*s2);
if(h<num){
ans=i;
num=h;
}
}
}else if(b1<b2){
int h;
num=b2-b1;
for(int i=1;i<m;++i){
h=abs(b1-b2+(m-i)*s2);
if(h<num){
ans=i;
num=h;
}
}
}
printf("%lld",ans);
return 0;
}