P5016 龙虎斗

        思路：定义一个一维数组，存储每个兵营所含的人数，假设这个数组为a数组,100%的数据满足 n≤10^5；所以 a[100010];然后，再求出龙虎双方的势力：

for(int i=1;i<m;++i)
2 {
    p+=(m-i)*a[i];
}
for(int i=m+1;i<=n;++i)
{
    q+=(i-m)*a[i];
}

        p代表龙方的势力，q代表虎方的势力，稳妥一些，开long long比较好；

        然后再求出两方的势力差（要正数，用abs函数取绝对值就可以）然后就是从1到n循环一次，循环时，我分成了三种情况：

for(int i=1;i<=n;++i)
    {
        if(i<m)
        {
            p+=s*(m-i);
            w=abs(p-q);
            if(w<t)
            {
                t=w;
                ans=i;
            }
            p=p1;
        }
        if(i==m)
        {
            w=abs(p-q);
            if(w<t)
            {
                t=w;
                ans=i;
            }
        }
        if(i>m)
        {
            q+=s*(i-m);
            w=abs(p-q);
            if(w<t)
            {
                t=w;
                ans=i;
            }
            q=q1;
        }
    }

        t为两方的势力差+1，p1和q1分别为p和q的备份。当i＜m时，就相当于增加龙方的势力，然后将它所增加的势力算出来，加到p上，再计算p与q的差，判断相比于之前是否差值变小，变小的话就将t更新为现在的差，并记录数组的位置。另外两种情况也类似。这样跑一遍，ans就是最终的答案。

        在此之前我还进行了一次特判，如果我手里的工兵为0的话，就直接输出1，因为没有工兵的话改变不了两方的差值，并且输出要求如果存在多个编号同时满足最优，取最小的编号。

完整代码：

#include<iostream>
#include<cmath>
using namespace std;
int long long a[100010];
int main()
{
    long long n,p=0,q=0,p1,q1,ans,w,t;
    cin>>n;
    for(int i=1;i<=n;++i)
    {
        cin>>a[i];
    }
    long long m,z,k,s;
    cin>>m>>z>>k>>s;
    if(s==0)
    {
        cout<<"1";
        return 0;
    }
    a[z]+=k;
    for(int i=1;i<m;++i)
    {
        p+=(m-i)*a[i];
    }
    for(int i=m+1;i<=n;++i)
    {
        q+=(i-m)*a[i];
    }
    p1=p,q1=q;
    t=abs(p-q)+1;
    for(int i=1;i<=n;++i)
    {
        if(i<m)
        {
            p+=s*(m-i);
            w=abs(p-q);
            if(w<t)
            {
                t=w;
                ans=i;
            }
            p=p1;
        }
        if(i==m)
        {
            w=abs(p-q);
            if(w<t)
            {
                t=w;
                ans=i;
            }
        }
        if(i>m)
        {
            q+=s*(i-m);
            w=abs(p-q);
            if(w<t)
            {
                t=w;
                ans=i;
            }
            q=q1;
        }
    }
    cout<<ans;
    return 0;
}