龙虎斗 C++

#include<bits/stdc++.h> 
using namespace std;
int main(){
    long long a[100000],b[100000]={0},m,s1,p1,s2,sum1=0,sum2=0,min=100000000000,flag=0,n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i]; 
    cin>>m>>p1>>s1>>s2;
    a[p1]+=s1;
    for(int i=1;i<=m-1;i++) sum1+=a[i]*(m-i); 
    for(int i=m+1;i<=n;i++) sum2+=a[i]*(i-m);
    if(sum1>sum2){
        long long t=sum1-sum2;
        for(int i=m+1;i<=n;i++){
            b[i]=(i-m)*s2+sum2;
            if(labs(b[i]-sum1)<min){
                flag=i;
                min=abs(b[i]-sum1);
            }
        }
        if(min>t) cout<<m;
        else cout<<flag;
    }
    else{
        long long t=sum2-sum1;
        for(int i=1;i<=m-1;i++){
            b[i]=(m-i)*s2+sum1;
            if(labs(b[i]-sum2)<min){
                flag=i;
                min=abs(b[i]-sum2);
            }
        }
        if(min>t) cout<<m;
        else cout<<flag;
    }
    return 0;     
}