LeetCode练习十六

1275.找出井字棋的获胜者

A 和 B 在一个 3 x 3 的网格上玩井字棋。

井字棋游戏的规则如下：

• 玩家轮流将棋子放在空方格 (" ") 上。
• 第一个玩家 A 总是用 "X" 作为棋子，而第二个玩家 B 总是用 "O" 作为棋子。
• "X" 和 "O" 只能放在空方格中，而不能放在已经被占用的方格上。
• 只要有 3 个相同的（非空）棋子排成一条直线（行、列、对角线）时，游戏结束。
• 如果所有方块都放满棋子（不为空），游戏也会结束。
• 游戏结束后，棋子无法再进行任何移动。

给你一个数组 moves，其中每个元素是大小为 2 的另一个数组（元素分别对应网格的行和列），它按照 A 和 B 的行动顺序（先 A 后 B）记录了两人各自的棋子位置。

如果游戏存在获胜者（A 或 B），就返回该游戏的获胜者；如果游戏以平局结束，则返回 "Draw"；如果仍会有行动（游戏未结束），则返回 "Pending"。

你可以假设 moves 都 有效（遵循井字棋规则），网格最初是空的，A 将先行动。

示例 1：

输入：moves = [[0,0],[2,0],[1,1],[2,1],[2,2]]
输出："A"
解释："A" 获胜，他总是先走。
"X  "    "X  "    "X  "    "X  "    "X  "
"   " -> "   " -> " X " -> " X " -> " X "
"   "    "O  "    "O  "    "OO "    "OOX"

示例 2：

输入：moves = [[0,0],[1,1],[0,1],[0,2],[1,0],[2,0]]
输出："B"
解释："B" 获胜。
"X  "    "X  "    "XX "    "XXO"    "XXO"    "XXO"
"   " -> " O " -> " O " -> " O " -> "XO " -> "XO " 
"   "    "   "    "   "    "   "    "   "    "O  "

示例 3：

输入：moves = [[0,0],[1,1],[2,0],[1,0],[1,2],[2,1],[0,1],[0,2],[2,2]]
输出："Draw"
解释：由于没有办法再行动，游戏以平局结束。
"XXO"
"OOX"
"XOX"

示例 4：

输入：moves = [[0,0],[1,1]]
输出："Pending"
解释：游戏还没有结束。
"X  "
" O "
"   "

提示：

• 1 <= moves.length <= 9
• moves[i].length == 2
• 0 <= moves[i][j] <= 2
• moves 里没有重复的元素。
• moves 遵循井字棋的规则。

class Solution:
    def tictactoe(self, moves: List[List[int]]) -> str:
        A_moves = []
        B_moves = []
        i = 0
        while i < len(moves):
            if i % 2 == 0:
                A_moves.append(moves[i])
            else:
                B_moves.append(moves[i])
            i += 1
        if [0,0] in A_moves and [1,1] in A_moves and [2,2] in A_moves :
            return "A"
        elif [0,2] in A_moves and [1,1] in A_moves and [2,0] in A_moves :
            return "A"
        elif [0,0] in B_moves and [1,1] in B_moves and [2,2] in B_moves :
            return "B"
        elif [0,2] in B_moves and [1,1] in B_moves and [2,0] in B_moves :
            return "B"
        else:
            count = [0, 0, 0, 0, 0, 0]
            for i in range(len(A_moves)):
                for j in range(len(A_moves[i])):
                    if j == 0:
                        a = A_moves[i][0]
                        count[a] += 1
                    elif j == 1:
                        b = A_moves[i][j]
                        count[b+3] += 1
            if max(count) == 3:
                return "A"
            else:
                count = [0, 0, 0, 0, 0, 0]
                for i in range(len(B_moves)):
                    for j in range(len(B_moves[i])):
                        if j == 0:
                            a = B_moves[i][0]
                            count[a] += 1
                        elif j == 1:
                            b = B_moves[i][j]
                            count[b+3] += 1
                if max(count) == 3:
                    return "B"
                elif len(moves)== 9:
                    return "Draw"
                else:
                    return "Pending"

可以用函数化简代码。

嵌套列表的使用。

循环写法。