洛谷 P5016 龙虎斗 2018 NOIP 普及组 T2

通过时间：2019.3.15
测评地址：https://www.luogu.org/problemnew/show/P5016
耗时/内存：111ms, 1180KB




代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[1000010];
long long temp,nx=999999999,n,i,m,p1,s1,s2;
long long lt=0,rt=0,ny;
int main()
{
	scanf("%lld",&n);
	for(i=1;i<=n;i++) scanf("%d",&a[i]);//scanf和cin差100ms！ 
	scanf("%lld %lld %lld %lld",&m,&p1,&s1,&s2);
	a[p1]+=s1;
	for(i=1;i<m;i++) lt+=a[i]*(m-i);
	for(i=m+1;i<=n;i++) rt+=a[i]*(i-m);
	temp=abs(rt-lt);
	if(lt>rt)
	{
		for(i=m+1;i<=n;i++)
			if( nx > abs((s2)*(i-m)-temp) ){
				nx=abs((s2)*(i-m)-temp);
				ny=i;
			}
		if(nx>temp) cout<<m;
		else cout<<ny;
	}
	if(lt==rt)cout<<m;
	if(lt<rt)
	{
		for(i=1;i<m;i++)
			if( nx > abs((s2)*(m-i)-temp) ){
				nx=abs((s2)*(m-i)-temp);
				ny=i;
			}
		if(nx>temp) cout<<m;
		else cout<<ny;
	}
	return 0;
}