八十分蒟蒻求助
查看原帖
八十分蒟蒻求助
373897
DreamBuilder楼主2021/6/7 21:02

最后五个点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;
}
2021/6/7 21:02
加载中...