【洛谷】P5016 [NOIP2018 普及组] 龙虎斗（详细代码）

1.【题目描述】

P5016 [NOIP2018 普及组] 龙虎斗

2.【代码】

#include<bits/stdc++.h>
using namespace std;
long long a[1000001];
long long n,m,p1,s1,s2;
long long fun1(long long x)
{
	long long sum1=0;
	for(long long i=1;i<x;i++)
	{
			sum1=sum1+a[i]*(x-i);
	}
	return sum1;
}
long long fun2(long long x)
{
	long long sum1=0;
	for(long long i=x+1;i<=n;i++)
	{
			sum1=sum1+a[i]*(i-x);
	}
	return sum1;
}
long long  fun3(long long x,long long sum2,long long sum1)
{
	if(x<m)
		sum1=sum1+s2*(m-x);
	else if(x>m)
		sum2=sum2+s2*(x-m);
	return abs(sum1-sum2);
}
int main()
{
	cin>>n;
	for(long long i=1;i<=n;i++)
		cin>>a[i];
	cin>>m>>p1>>s1>>s2;
	a[p1]=a[p1]+s1;//给p1兵营增加工兵 
	long long sum1=fun1(m);//计算龙的气势 
	long long sum2=fun2(m);//计算虎的气势 
	long long min=999999999,j=1;
	for(long long i=1;i<=n;i++)
	{
		long long s=fun3(i,sum2,sum1);//寻找最小值 
		if(min>s)
		{
			min=s;
			j=i;//记录最小值的位置 
		}
	}
	cout<<j;
	return 0;
}

仅供参考！