(Python)五子棋迷

张兵和王武是五子棋迷，工作之余经常切磋棋艺。
走了一会儿，轮到张兵了，他对着一条线思考起来了，这条线上的棋子分布如下：
用数组表示：
-1 0 1 1 1 0 1 0 1 -1
棋子分布说明:
-1代表白子， 0代表空位， 1代表黑子；
数组长度L, 满足1 < L < 40, L为奇数；
请帮他写一个程序，算出最有利的出子位置。
最有利定义：
找到一个空位0，用棋子 (1/-1) 填充该位置，可以使得当前子的最大连续长度变大；
如果存在多个位置，返回最靠近中间的较小的那个坐标；
如果不存在可行位置，直接返回 -1；
连续长度不能超过 5 个(五子棋约束)；
输入描述
第一行: 当前出子颜色
第二行: 当前的棋局状态
输出描述
1 个整数，表示出子位置的数组下标

示例一：
输入：
1
-1 0 1 1 1 0 1 0 1 -1 1
输出：
5
当前为黑子（1），放置在下标为 5 的位置，黑子的最大连续长度，可以由 3 到 5。

示例二：
输入：
-1
-1 0 1 1 1 0 1 0 1 -1 1
输出：
1
当前为白子，唯一可以放置的位置下标为 1， 白子的最大长度，由 1 变为 2。

示例三：
输入：
1
0 0 0 0 1 0 0 0 0 1 0
输出：
5
可行的位置很多，5 最接近中间的位置坐标。

def find_best_move(target, nums):
    max_continuous_length = 0  # 记录最大连续长度
    best_target = -1  # 记录最有利的出子位置下标
    for i in range(len(nums)):
        if nums[i] == 0:  # 只考虑空位
            # 在当前位置放置棋子
            nums[i] = target
            # print(f"nums:{nums}")
            # 计算当前位置的连续长度（只需考虑当前位置的左侧和右侧）
            continuous_length = 1  # 初始长度为1
            for l in [-1, 1]:  # 这个循环会执行两次，分别对应着检查左侧和右侧的方向
                j = i + l  # j 表示当前位置的相邻位置，l 控制方向，-1 表示左侧，1 表示右侧
                while 0 <= j < len(nums) and nums[j] == target:
                    # 这是一个 while 循环，它会一直执行直到不满足下列条件之一：
                    # 1. j 超出了棋盘边界，即 j 小于 0 或大于等于棋盘大小
                    # 2. j 处的棋子颜色与当前颜色不相同
                    # 如果当前位置 j 处的棋子颜色与当前颜色相同，就增加连续长度
                    continuous_length += 1
                    j += l
            mid = len(nums) // 2
            if continuous_length > max_continuous_length or continuous_length == max_continuous_length:
                max_continuous_length = continuous_length
                if abs(i - mid) < abs(best_target - mid):
                    best_target = i
    return best_target


if __name__ == '__main__':
    target = int(input())
    chess = list(map(int, input().split()))
    best_target = find_best_move(target, chess)
    print(best_target)