CSP-J-2018 龙虎斗 100%分数正解

题目描述

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有n个兵营（自左向右编号1~n），相邻编号的兵营之间相隔1厘米，即棋盘长度为n-1厘米的线段。i号兵营里有ci位工兵。
    下面图1为n=6的示例：

    轩轩在左侧，代表“龙”；凯凯在右侧，代表“虎”。他们以m号兵营作为分解，靠左的工兵属于龙势力，靠右的工兵属于虎势力，而第m号兵营中的工兵很纠结，他们不属于任何一方。
    一个兵营的气势为：该兵营中工兵数 X 该兵营到m号兵营的距离；参与游戏的一方的势力定义为：属于这一方所有兵营的气势之和。
    下面图2为n=6，m=4的示例，其中红色为龙方，黄色为虎方：

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

 输入

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

输出

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

样例输入1

6
2 3 2 3 2 3
4 6 5 2

样例输出1

2

样例输入2 

6
1 1 1 1 1 16
5 4 1 1

样例输出2

1

看正解：

#include<bits/stdc++.h>
using namespace std;
int main(){
    long long int a[100010],tiger=0,dragon=0,min=1000000,x,s1,s2,index=0;
    int n,m,p1,p2;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    cin>>m>>p1>>s1>>s2;
    a[p1]+=s1;
    for(int i=1;i<=n;i++){
        if(i>m){
            tiger+=a[i]*(i-m);
        }else if(i<m){
            dragon+=a[i]*(m-i);
        }
    }
    min=labs(dragon-tiger);
    if(dragon>tiger){
        p2=m;
        for(int i=m+1;i<=n;i++){
            x=s2*(i-m);
            index=labs(dragon-(tiger+x));
            if(index<min){
                min=index;
                p2=i;
            }
        }
    }else if(dragon==tiger){
        p2=m;
    }else if(dragon<tiger){
        p2=m;
        for(int i=1;i<m;i++){
            x=s2*(m-i);
            index=labs(tiger-(dragon+x));
            if(index<min){
                min=index;
                p2=i;
            }
        }
    }
    cout<<p2;
    return 0;
}

 

 拜拜~~~下期见！