80分WA
查看原帖
80分WA
158878
B1ade_楼主2020/11/6 16:28
#include<bits/stdc++.h>
using namespace std;
long long n,c[100005],m,p1,s1,s2,qs[100005];
int main()
{
	cin>>n;
	for (int i=1;i<=n;++i) cin>>c[i];
	cin>>m>>p1>>s1>>s2;
	long long l=0,h=0;
	for (int i=1;i<=n;++i)
	{
		if (i==p1)
		{
			c[i]+=s1;
		}
		qs[i]=abs(i-m)*c[i];
		if (i<m) l+=qs[i];
		if (i>m) h+=qs[i];
	}
	if (l==h)
	{
		cout<<m;
		return 0;
	}
	long long p2=m,cha=abs(l-h);
	for (int i=1;i<=n;++i)
	{
		int th=h,tl=l;
		if (i<m) tl+=(m-i)*s2;
		else th+=(i-m)*s2;
		if (abs(tl-th)<=cha)
		{
			p2=i;
			cha=abs(tl-th);
		}
	}
	cout<<p2;
	return 0;
}
2020/11/6 16:28
加载中...