【NOIP普及组】1979：【18NOIP普及组】龙虎斗

)

1.【题目描述】

【题目描述】
轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有n个兵营（自左至右编号1∼n），相邻编号的兵营之间相隔 1 厘米，即棋盘为长度为n−1厘米的线段。i号兵营里有ci位工兵。

下面图1为n=6的示例：

图 1. n=6的示例

轩轩在左侧，代表“龙”；凯凯在右侧，代表“虎”。他们以m号兵营作为分界，靠左的工兵属于龙势力，靠右的工兵属于虎势力，而第m号兵营中的工兵很纠结，他们不属于任何一方。

一个兵营的气势为：该兵营中的工兵数×该兵营到m号兵营的距离；参与游戏一方的势力定义为：属于这一方所有兵营的气势之和。

下面图2为n=6,m=4的示例，其中红色为龙方，黄色为虎方：

图 2. n=6,m=4的示例

游戏过程中，某一刻天降神兵，共有s1位工兵突然出现在了p1号兵营。作为轩轩和凯凯的朋友，你知道如果龙虎双方气势差距太悬殊，轩轩和凯凯就不愿意继续玩下去了。为了让游戏继续，你需要选择一个兵营p2，并将你手里的s2 位工兵全部派往兵营p2，使得双方气势差距尽可能小。

注意：你手中的工兵落在哪个兵营，就和该兵营中其他工兵有相同的势力归属（如果落在m号兵营，则不属于任何势力）。

【输入】
输入的第一行包含一个正整数n，代表兵营的数量。

接下来的一行包含n个正整数，相邻两数之间以一个空格分隔，第i个正整数代表编号为i的兵营中起始时的工兵数量ci。

接下来的一行包含四个正整数，相邻两数间以一个空格分隔，分别代表m,p1,s1,s2。

【输出】
输出有一行，包含一个正整数，即p2，表示你选择的兵营编号。如果存在多个编号同时满足最优，取最小的编号。

【输入样例】
6
2 3 2 3 2 3
4 6 5 2
【输出样例】
2

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;
        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;
}

仅供参考！