五子棋迷 Python3

张兵和王武是五子棋迷，工作之余经常切磋棋艺。这不，这会儿又下起来了。走了一会儿，轮张兵了，对着一条线思考起来了，这条线上的棋子分布如下

用数组表示: -1 0 1 1 1 0 1 01 1

棋子分布说明:

1.-1代表白子，0代表空位，1 代表黑子

2.数组长度L,满足 1 < L < 40,且L为奇数

你得帮他写一个程序，算出最有利的出子位置。最有利定义

1.找到一个空位(0)，用棋子(1/-1)填充该位置，可以使得当前子的最大连续长度变大

2.如果存在多个位置，返回最靠近中间的较小的那个坐标;

3.如果不存在可行位置，直接返回-1:

4.连续长度不能超过5个(五字棋约束)

输入描述:

第一行: 当前出子颜色

第二行: 当前的棋局状态

输出描述

1个整数，表示出子位置的数组下标

本地代码输入方式跟acm有所不同，我是用一个文本文档存储测试用例的

file_path=r'C:\test.txt'
file =open(file_path,'r',encoding='utf-8')
mlist=[line for line in file]
for i in range(0,len(mlist),2):
	cur=mlist[i].rstrip()
	lt=mlist[i+1].split()
	n=len(lt)
	l1=[0]*n
	r1=[0]*n
	cnt=0
	#列表l1统计每个0位置左边同颜色棋子数目
	for j in range(n):
		if lt[j]==cur:
			cnt=cnt+1
		else:
			if lt[j]=='0':
				l1[j]=cnt
			cnt=0
	cnt=0
	oldMax=0
	#列表r1统计每个0位置右边同颜色棋子数目
	for j in range(n-1,-1,-1):
		if lt[j]==cur:
			cnt=cnt+1
		else:
			oldMax=max(oldMax,cnt)
			if lt[j]=='0':
				r1[j]=cnt
			cnt=0
	#列表suml统计在0位置下棋后，连续棋子长度
	suml=list(map((lambda x,y:x+y+1),l1,r1))
	symbol=1
	p=n//2
	newMax=0
	newMax_index=p
	#从中间开始往两边交替遍历
	for j in range(n):
		p+=j*symbol
		if suml[p]>newMax:
			newMax=suml[p]
			newMax_index=p
		symbol*=-1
	print(newMax_index) if newMax>oldMax else print(-1)