← Back to List

34386번: Tic-Tac-Toe AI ↗

Solutions

Python 3
2.2 KB | 2222 chars
"""
[34386: Tic-Tac-Toe AI](https://www.acmicpc.net/problem/34386)

Tier: Bronze 2
Category: implementation, bruteforcing, case_work
"""


import sys
from math import sqrt, pi, sin, factorial, ceil, floor
from datetime import datetime, timedelta
from collections import deque, defaultdict, Counter
from itertools import permutations, combinations, product
from bisect import bisect_left, bisect_right
from heapq import heappush, heappop, heapify
from functools import reduce, lru_cache
from operator import itemgetter, attrgetter, mul, add, sub, truediv
from typing import List, Tuple, Dict, Set, Any, Union
from fractions import Fraction

SYS_INPUT = True
RECURSION_LIMIT = 10 ** 7
SET_RECURSION = False
BLANK = " "

if SET_RECURSION:
  sys.setrecursionlimit(RECURSION_LIMIT)

inp = lambda : sys.stdin.readline().rstrip() if SYS_INPUT else input()
mii = lambda : [*map(int,inp().split())]
mfi = lambda : [*map(float,inp().split())]
ii = lambda : int(inp())
fi = lambda : float(inp())
isplit = lambda : inp().split()
p = print

def gcd(a, b): return gcd(b, a % b) if b > 0 else a
def lcm(a, b): return a * b // gcd(a, b)
def near_integer(x): return int(x + 0.5) if x >= 0 else int(x - 0.5)
def round_up_half(n): return floor(n + 0.5)
def rotate90(l): return [''.join(x) for x in zip(*l[::-1])]
def transpose(matrix): return list(map(list, zip(*matrix)))
def arithmetic_seq_sum(a, n, d): return (a * n + n * (n - 1) * d  // 2)


def is_won(player, grid):
  if sum([int(grid[i][i] == player) for i in range(3)]) == 3:
    return True

  if sum([int(grid[2 - i][i] == player) for i in range(3)]) == 3:
    return True
  
  for i in range(3):
    if sum([int(grid[i][j] == player) for j in range(3)]) == 3:
      return True
    
    if sum([int(grid[j][i] == player) for j in range(3)]) == 3:
      return True
  
  return False


def solve():
  player = inp()  
  grid = [list(inp()) for _ in range(3)]
  
  for i in range(3):
    for j in range(3):
      if grid[i][j] != 'E':
        continue
      
      grid[i][j] = player
        
      if is_won(player, grid):
        print(i + 1, j + 1)
        return
      
      grid[i][j] = 'E'


if __name__ == "__main__":
  tc = 1
  for t in range(1, tc+1):
    ret = solve()