编程练习之龙虎斗题

题目描述：

轩轩和开开正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有n个兵营(自左至右编号1~n)，相邻编号的兵营之间相隔1厘米，即棋盘为长度为n-1厘米的线段。i号兵营里有ci位工兵。

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

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

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

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

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

输入描述：

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

输出描述：

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

个人代码：

import java.util.Scanner;
/**
 * 牛客网题目链接：https://ac.nowcoder.com/acm/contest/277/B
 * AC情况：只通过60%
 * @author HPvWu
 * @time 2019-1-3
 */
public class DragonTigerFight {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int[] a = new int[9999999];
		for(int i=1;i<=n;i++) {
			a[i] = scanner.nextInt();
		}
		long m = scanner.nextInt();
		long p = scanner.nextInt();
		long s1 = scanner.nextInt();
		long s2 = scanner.nextInt();
		long leftsum = 0;
		long x = m;
		a[(int) (p)] += s1;
		for(int i=1;i<m;i++) {
			leftsum += a[i] * Math.abs(i-m);
		}
		int rightsum = 0;
		for(int i=n;i>m;i--) {
			rightsum += a[i] * Math.abs(i-m);
		}
		long min = Math.abs(rightsum-leftsum);
		if(leftsum < rightsum) {
			for(int i=1;i<m;i++) {
				if(Math.abs(rightsum-leftsum-(m-i)*s2)<min) {
					min = Math.abs(rightsum-leftsum-(m-i)*s2);
//					System.out.println(min);
					x = i;
				}
			}
		}else if(leftsum > rightsum) {
			for(long i=m+1;i<n;i++) {
				if(Math.abs(leftsum-rightsum-(i-m)*s2) < min) {
					min = Math.abs(leftsum-rightsum-(i-m)*s2);
					System.out.println(min);
					x = i;
				}
			}
		}else {
			x = m;
		}
		System.out.println(x);
	}
}

 

问题：

该代码仍不能完全通过测试，望各位大佬看见代码段之后给我些修改提示，使该代码能成功AC，谢谢各位。