P5016 龙虎斗 题解

这题真是*到家了QAQ

我在考场上调了将近75min，总算过了大样例。

首先，我们可以简化这一题，这道题的本质就是让我们找出一个点p2，往那个点上面加上s2个单位的重量，使得以m为中的两边的权值和的差尽量的小。

其中权值和的计算方法是：对于第i号节点（i！=m），令m_i为它的权值，则有m_i=|i-m|*a_i,其中a_i为输入时的该兵营的兵力总量。

明白了这一点后，剩下的就好办了。我们只要用一个bl_l来存储轩轩一方的权值总和，用bl_r来存储凯凯一方的权值总和；（注意，要用bl_l和bl_r 要开unsigned long long）

最后，计算cha = |bl_l-bl_r|;再用cha 来除以输入的s2（你手中拥有的兵力），就是answer 了

此时，还有一个要注意的地方，如果answer超过了n或者小于了1，那么就要相应地将其变成n或者1，因为答案必须在1~n之间。

而且answer需要四舍五入。

下面是代码：

#include <cstdio>
#include <algorithm>
using namespace std;
#define ll unsigned long long
int a[100005],n,m,p1,s1,s2;
ll bl_l,bl_r;
int main()
{
//    freopen("fight.in","r",stdin);
//    freopen("fight.out","w",stdout);
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    scanf("%d %d %d %d",&m,&p1,&s1,&s2);
    a[p1]+=s1;
    for (int i=1;i<m;i++) bl_l+=(ll)(m-i)*a[i];
    for (int i=(int)m+1;i<=n;i++) bl_r+=(ll)(i-m)*a[i];
//    printf("%llu %llu\n",bl_l,bl_r);
    if (bl_l==bl_r) printf("%d\n",m);
    else if (bl_l<bl_r)
    {
        ll cha=bl_r-bl_l;
//        printf("%lld\n",cha);
        double mm=(double)cha/(double)s2;
//        printf("%lf\n",mm);
        ll dis_m=(mm*10+5)/10;
        if (m-dis_m<=0) printf("1\n");
        else printf("%lld\n",m-dis_m);
    }
    else
    {
        ll cha=bl_l-bl_r;
//        printf("%lld\n",cha);
        double mm=(double)cha/(double)s2;
//        printf("%lf\n",mm);
        ll dis_m=(mm*10+5)/10;
        if (m+dis_m>n) printf("%d\n",n);
        else printf("%lld\n",m+dis_m);
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}