P5016 [NOIP2018 普及组] 龙虎斗（C语言）

题目见https://www.luogu.com.cn/problem/P5016

代码：

#include <stdio.h>
long long a[100005],n,m,p,s1,s2,sum=0,ret;
int main()
{
	int i;
	scanf("%lld",&n);
	for(i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
	}
	scanf("%lld%lld%lld%lld",&m,&p,&s1,&s2);
	for(i=1;i<=n;i++)//直接以正负表示两边势力差距，越接近0差距越小; 
	{
		sum+=a[i]*(m-i);
	}
	sum+=s1*(m-p);//加上天降神兵; 
	ret=m+(int)(sum*1.0/s2+0.5*(sum>0?1:-1));
//直接求势力差为0时需把s2个兵放哪，取最接近该数的整数，
//如2.1取2,2.8取3;   0.5三目运算为进位操作，因为强制转换只取整数部分; 
	if(ret<0)
	{
		ret=1;
	}
	if(ret>n)
	{
		ret=n;
	}
	printf("%lld\n",ret);
}