CSP-J 2018 T2 龙虎斗

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+20;
int n,a[N],m,p1,s1,s2,num1,num2,t1,t2,minn=0x3f3f3f3f,ans;
int f(int x,int y){
    if(x>=y) return x-y;
    return y-x;
}
signed main(){
    freopen("fight.in","r",stdin);
    freopen("fight.out","w",stdout);
    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<=n;i++){
        if(i<m){
            num1+=(m-i)*a[i];
        }else if(m<i){
            num2+=(i-m)*a[i];
        }
    }
    for(int i=1;i<=n;i++){
        t1=num1;
        t2=num2;
        if(i<m){
            t1+=(m-i)*s2;
        }else if(m<i){
            t2+=(i-m)*s2;
        }
        int t3=f(t1,t2);
        if(minn>t3){
            minn=t3;
            ans=i;
        }
    }
    cout<<ans;
    return 0;
}