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;
}
仅供参考!