80WA求助
查看原帖
80WA求助
1264576
2010zhuanghaojun楼主2024/6/20 13:59
#include<bits/stdc++.h>
using namespace std;
long long a[100001];
int main(){
    int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	} 
	int m,p1,s1,s2;
	cin>>m>>p1>>s1>>s2;
	a[p1]+=s1;
	long long l=0,h=0;
	for(int i=1;i<m;i++){
		l+=(m-i)*a[i];
	}
	for(int i=m+1;i<=n;i++){
		h+=(i-m)*a[i];
	}
	//cout<<l<<" "<<h<<endl;
	long long minn=abs(h-l),p2=m;
	for(int i=1;i<=m;i++){
		long long l1=l;
		l1+=s2*(m-i);
		if(abs(h-l1)<minn){
		    p2=i;
			minn=abs(h-l1);
		}	
	}
	for(int i=m;i<=n;i++){
		long long h1=h;
		h1+=s2*(i-m);
		if(abs(h1-l)<minn){
		    p2=i;
			minn=abs(h1-l);	
		}	
	}
	cout<<p2;
}
2024/6/20 13:59
加载中...