C++星际问题03：【模拟】龙虎斗

题目描述：

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

轩轩在左侧，代表“龙”；凯凯在右侧，代表“虎”。他们以m号兵营作为分界，靠左的工兵属于龙势力，靠右的工兵属于虎势力，而第m号兵营中的工兵很纠结，他们不属于任何一方。
一个兵营的气势为：该兵营中的工兵数×该兵营到m号兵营的距离；参与游戏一方的势力定义为：属于这一方所有兵营的气势之和。
下面图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

解题思路：

使用LLONG_MAX函数和abs函数求出p2的值并输出

代码：

#include <bits/stdc++.h>
#define MX 100000
using namespace std;
int main()
{
	int n,i;
	long long int m,p1,p2,s1,s2,tiger = 0,dragon = 0,c[MX + 10];
	cin >> n;
	for(int i = 1;i <= n;i++)
	{
		cin >> c[i];
	}
	cin >> m >> p1 >> s1 >> s2;
	c[p1] += s1;
	long long int mn = LLONG_MAX;
	for(int i = 1;i <= m - 1;i++)
	{
		dragon = dragon + c[i] * (m - i);
	}
	for(int i = m + 1;i <= n;i++)
	{
		tiger = tiger + c[i] * (i - m);
	}
	for(int i = 1;i <= n;i++)
	{
		c[i] = c[i] + s2;
		if(mn > abs(dragon + s2 * (m - i) - tiger))
		{
			mn = abs(dragon + s2 * (m - i) - tiger);
			p2 = i;
		}
	}
	cout << p2 << endl;
	return 0;
}

注意，这里使用的是万能头文件，可能您的编译器不受支持。如不受支持，请改为iostream头文件