92分求助
查看原帖
92分求助
554533
wangtiantian72楼主2021/10/8 17:09

7,8WA,求助

#include <bits/stdc++.h>
using namespace std;
long long n,s1,s,p1,p,a[100001],h,l,m,j,mi=1000000000;
int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    cin>>m>>p1>>s1>>s;
    a[p1]+=s1;
    for(int i=1;i<m;i++) l+=a[i]*(m-i);
    for(int i=m+1;i<=n;i++)h+=a[i]*(i-m);
    j=abs(l-h);
    if(l==h||s==0){
        cout<<m;
        return 0;
    }
    if(l<h){
        for(int i=1;i<=m;i++){
            j=abs(l-h);
            j=abs(j-s*(m-i));
            if(j<mi){
				mi=j;
				p=i;
            }
        }
    }
    else{
        for(int i=m+1;i<=n;i++){
            j=abs(l-h);
            j=abs(j-s*(i-m));
            if(j<mi){
                mi=j;
                p=i;
            }
        }   
    }
    cout<<p;
    return 0;
}

救救我orz

2021/10/8 17:09
加载中...