Строки-2 "Три в ряд" (lms yandex)

n = int(input())
pol = "_" * n
pol = " ".join(pol).split(" ")
pols = pol.copy()
k = 1
s = 0
score1 = 0
score2 = 0
while n:
    h1 = input()
    h1 = h1.split(" ")
    if k % 2 == 1:
        if int(h1[0]) < 0 or int(h1[0]) >= n or pol[int(h1[0])] != "_":
            print("This place is taken.")
        else:
            pol[int(h1[0])] = h1[1]
            print("AI step", " ".join(h1))
            k += 1
    else:
        if int(h1[0]) < 0 or int(h1[0]) >= n or pol[int(h1[0])] != "_":
            print("This place is taken.")
        else:
            pol[int(h1[0])] = h1[1]
            print("Your step", " ".join(h1))
            k += 1
    for i in range(len(pol) - 2):
        if pol[i] == pol[i + 1] == pol[i + 2] and pol[i] != "_":
            print(*pol)
            pol[i] = '_'
            pol[i + 1] = '_'
            pol[i + 2] = '_'
            if k % 2 == 0:
                score2 += 1
            elif k % 2 == 1:
                score1 += 1
    print(*pol)
    for i in pol:
        if i != "_":
            s += 1
    if s == n:
        if k % 2 == 0 and score2 != score1:
            print("You win!", score1, ":", score2)
            break
        elif k % 2 == 1 and score2 != score1:
            print("Ai win!", score2, ":", score1)
            break
        else:
            print("We have a tie.")
            break

from random import choice, shuffle
 
def find_3(a):
    for i in range(len(a) - 3 + 1):
        s = a[i:i + 3]
        if s.count('_') == 1 and len(se := set(s)) == 2:
            se.remove('_')
            return s.index('_') + i, se.pop()
 
def is_good(a):
    return not find_3(a)
 
def remove_3(a):
    for C in 'RGB':
        if C*3 in (s := ''.join(a)):
            s = s.replace(C*3, '___')
            return list(s)
    return a
 
def get_step(a):
    if (res := find_3(a)):
        return res
    space_ind = [i for i in range(len(a)) if a[i]=='_']
    shuffle(space_ind)
    for i in space_ind:
        a_cur = list('RGB')
        shuffle(a_cur)
        for color in a_cur:
            a_new = a.copy()
            a_new[i] = color
            if is_good(a_new):
                return i, color
    return choice(space_ind), choice(a_cur)
 
n = int(input('Ширина поля->'))
arr = list('_' * n)
res_comp = 0
res_user = 0
while True:
    i, color = get_step(arr)
    print(f'\nХод компьютера: {i} {color}')
    arr[i] = color
    print(*arr)
    arr_new = remove_3(arr)
    if arr_new != arr:
        res_comp += 1
        arr = arr_new
        print(*arr)
    if '_' not in arr:
        break
 
    while True:
        try:
            i, color = input('\nВаш ход->').split()
            i = int(i)
            if arr[i] != '_':
                print('Это место занято')
                continue
        except:
            continue
        if color in 'RGB':
            arr[i] = color
            break
    print(*arr)
    arr_new = remove_3(arr)
    if arr_new != arr:
        res_user += 1
        arr = arr_new
        print(*arr)
    if '_' not in arr:
        break
if res_comp > res_user:
    print(f'Компьютер победил! {res_comp} : {res_user}')
elif res_user > res_comp:
    print(f'Вы победили! {res_user} : {res_comp}')
else:
    print(f'Ничья. {res_user} : {res_comp}')