华为OD机试 Python 【下棋位置】

题目

张兵和王武喜欢玩五子棋。现在轮到张兵了，他面前的棋盘上有一排棋子。

棋子规则：
-1 表示白子
0 表示没子，是个空位
1 表示黑子

一排棋子中，棋子数量L要满足：1 < L < 40，并且L是奇数。

你要写个程序帮张兵找到最佳的落子位置。怎么判断“最佳”呢？

找个空位（0）落子，使得这种颜色的棋子连在一起的数量最多。

如果有多个这样的空位，选中间的那个。
但是，连在一起的棋子数量不能超过5个。

输入：

第一行：当前要下的棋子颜色（1或-1）
第二行：当前棋盘上的棋子排列
输出：

一个数字，表示落子位置的下标（如果没有好位置，输出-1）
例如：

输入：
1
-1 0 1 1 1 0 1 -1 1

输出：

5

代码

def optimal_move(piece_type, chess_line):