我是酷叮猫少儿编程的郭老师，今天我将为大家提供一道题，来自2018年NOIP的普及组复赛考试题《龙虎斗》. 下面让我们来看一下这道题.
[输入格式] 输入文件名为fight.in。 输入文件的第一行包含一个正整数 n，代表兵营的数量。 接下来的一行包含 n 个正整数，相邻两数之间以一个空格分隔，第 i 个正整数代 表编号为 i 的兵营中起始时的工兵数量 ci。 接下来的一行包含四个正整数，相邻两数间以一个空格分隔，分别代表 m, p1, s1, s2。
[输出格式] 输出文件名为fight.out。 输出文件有一行，包含一个正整数，即 p2，表示你选择的兵营编号。如果存在多 个编号同时满足最优，取最小的编号。
[输入输出样例1]
fight.in fight.out
6 2
2 3 2 3 2 3
4 6 5 2
[输入输出样例1 说明] 见问题描述中的图2。 双方以 m = 4 号兵营分界，有 s1 = 5 位工兵突然出现在 p1 = 6 号兵营。
龙方的气势为： 2 × (4 − 1) + 3 × (4 − 2) + 2 × (4 − 3) = 14
虎方的气势为： 2 × (5 − 4) + (3 + 5) × (6 − 4) = 18
当你将手中的 s2 = 2 位工兵派往 p2 = 2 号兵营时，
龙方的气势变为： 14 + 2 × (4 − 2) = 18 此时双方气势相等。
[输入输出样例2]
fight.in fight.out
6 1
1 1 1 1 1 16
5 4 1 1
[输入输出样例2 说明] 双方以 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 此时可以使双方气势的差距最小。
[程序代码]
# coding:utf-8
n=m=p1=s1=s2=None
soldiers=None
with open("fight.in", 'r') as f:
n = f.readline().strip()
n = int(n)
soldiers = [int(x) for x in f.readline().strip().split()]
m,p1,s1,s2 = [int(x) for x in f.readline().strip().split()]
# 求p2的兵营编号
def d_momentum(ss):
'''龙气势'''
dragon = 0
for sid, c in enumerate(ss,start=1):
if sid<m:
dragon += c*(m-sid)
return dragon
def t_momentum(ss):
'''虎气势'''
tiger = 0
for sid, c in enumerate(ss,start=1):
if sid>m:
tiger += c*(sid-m)
return tiger
def app():
news = soldiers[:]
news[p1-1] += s1 # p1突降士兵
dm = d_momentum(news)
tm = t_momentum(news)
if dm == tm:
return m
elif dm>tm:
cha = dm - tm
chalist = {}
for sid, c in enumerate(news[m+1:], start=m+1):
temp = abs(cha-s2*(sid-m))
# chalist[temp] = sid
if temp in chalist:
chalist[temp] += [sid]
else:
chalist[temp] = [sid]
mincha = min(chalist)
return min(chalist[mincha])
else:
cha = tm - dm
chalist = {}
for sid, c in enumerate(news[:m], start=1):
temp = abs(cha-s2*(m-sid))
if temp in chalist:
chalist[temp] += [sid]
else:
chalist[temp] = [sid]
mincha = min(chalist)
return min(chalist[mincha])
print(app()) # 打印结果
with open("fight.out", 'w') as f:
f.write("{}".format(app()))
