张兵和王武是五子棋迷,工作之余经常切磋棋艺。
走了一会儿,轮到张兵了,他对着一条线思考起来了,这条线上的棋子分布如下:
用数组表示:
-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)