题目大意:一条线段,共有n个点,m点左边是A的势力,右边是B的势力,你需要给任意位置加势力,在哪里加两边势力最接近?
题目描述
轩轩在左侧,代表“龙”;凯凯在右侧,代表“虎”。 他们以 m 号兵营作为分界, 靠左的工兵属于龙势力,靠右的工兵属于虎势力,而第m号兵营中的工兵很纠结,他 们不属于任何一方。
一个兵营的气势为:该兵营中的工兵数 × 该兵营到 m 号兵营的距离;参与游戏 一方的势力定义为:属于这一方所有兵营的气势之和。
下面图 2 为 n = 6, m = 4 的示例,其中红色为龙方,黄色为虎方:
输入
输入文件的第一行包含一个正整数 n,代表兵营的数量。
接下来的一行包含 n 个正整数,相邻两数之间以一个空格分隔,第 i 个正整数代表编号为 i 的兵营中起始时的工兵数量 ci。接下来的一行包含四个正整数,相邻两数间以一个空格分隔,分别代表 m, p1, s1, s2。
输出
输出文件有一行,包含一个正整数,即 p2,表示你选择的兵营编号。如果存在多个编号同时满足最优,取最小的编号。
样例输入
样例1:
6
2 3 2 3 2 3
4 6 5 2
样例2:
6
1 1 1 1 1 16
5 4 1 1
样例输出
样例1:2
样例2:1
提示
双方以 m=5 号兵营分界,有 s1 =1 位工兵突然出现在 P1 =4 号兵营。 龙方的气势为:1 × (5 − 1) + 1 × (5 − 2) + 1 × (5 − 3) + (1 + 1) × (5 − 4) = 11
虎方的气势为:16 × (6 − 5) = 16
当你将手中的 S2 = 1 位工兵派往 P2 = 1 号兵营时,龙方的气势变为:11 + 1 × (5 − 1) = 15
此时可以使双方气势的差距最小。
【数据规模与约定】
。
对于 的数据,。
另有 的数据,。
对于 的数据,。
对于 的数据,。
对于 的数据,,。
解题思路
模拟:读入数据,分别计算左右两边的势力,当然,天降神兵也可以先出来,就是稍微改一下初始数据而已。
枚举和查找:如果势力相等,放到m;如果左边势力大,放到右边,查找最小差的且标号最小的;如果左边势力小,放在左边,同样是查找最小差且编号小的。
注意:最后两组数据很大,会爆int,所以用long long。