题解 P5016 【龙虎斗】

这道题其实并不是很难，只需要暴力枚举就行了，应该是O(n)的复杂度

简单思路:暴力枚举每一个点后两方的气势差，但是，CCF会卡long long，所以我在考场上调试了很久才发现这个问题。
变量定义：nong为龙方的气势值，hu为虎方的气势值
贴上代码：

#include <iostream>
#include <cstdio> 
#include <cmath>
using namespace std;
long long a[100011];
int main()
{
	long long nong=0,hu=0;
	int i,j,k,n;
	long long m,s1,p1,p2;
	scanf("%d",&n);
	for(i=1;i<=n;i++) scanf("%lld",&a[i]);
	scanf("%lld %lld %lld %lld",&m,&p1,&s1,&p2);
	a[p1]+=s1;   
	for(i=1;i<=n;i++)
	{
		if(i<m) nong+=a[i]*(m-i);   //统计龙方的气势值
		else if(i>m) hu+=a[i]*(i-m); //统计虎方的气势值
	}
	long long ans=abs(nong-hu);    //ans初始化为将工兵放在m号兵营里面
	int step=m;   //最小号兵营一开始赋值为m号兵营
	for(i=1;i<=n;i++)
	{
		if(i<m) //在m号兵营左边加龙方的气势值
		{
			if(ans>abs((nong+p2*(m-i)-hu)))
			{
				ans=abs((nong+p2*(m-i)-hu));
				step=i;
			}
		}
		else if(i>m) //在m号兵营左边加龙方的气势值
		{
			if(ans>abs(nong-(hu+p2*(i-m))))
			{
				ans=abs(nong-(hu+p2*(i-m)));
				step=i; 
			}
		}
	}
	cout<<step;
	return 0;
}