【力扣 1275】找出井字棋的获胜者 C++题解（位集合+位运算+模拟）

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 遵循井字棋的规则。

---

思路

首先定义两个位集a和b，分别表示玩家A和玩家B在棋盘上的位置。然后，函数遍历moves中的每一个步骤，根据步骤的序号决定是玩家A还是玩家B进行移动，并根据步骤的内容更新相应的位集。每次更新位集后，函数都会检查该位集是否满足获胜条件。

获胜条件的检查是通过一个名为check的函数实现的。这个函数接受一个位集作为输入，然后遍历所有可能的获胜条件（即所有可能的三连线），并检查输入的位集是否满足其中的任何一个。如果满足，那么函数返回true，否则返回false。

在遍历完所有的步骤后，如果还没有玩家获胜，那么函数将根据步骤的数量判断比赛的结果。如果步骤的数量是10，那么所有的位置都已经被占用，比赛为平局，函数返回"Draw"。否则，还有位置没有被占用，比赛尚未结束，函数返回"Pending"。

---

AC代码

/*
 * @lc app=leetcode.cn id=1275 lang=cpp
 *
 * [1275] 找出井字棋的获胜者
 */

// @lc code=start
class Solution {
	const int masks[8] = {0b111000000, 0b000111000, 0b000000111, 0b100100100,
						  0b010010010, 0b001001001, 0b100010001, 0b001010100};

   public:
	string tictactoe(vector<vector<int>>& moves) {
		bitset<9> a, b;
		int cnt = 1;
		for (vector<int> move : moves) {
			int index = move[0] * 3 + move[1];
			if (cnt % 2) {
				a.set(index);
				if (check(a)) {
					return "A";
				}
			} else {
				b.set(index);
				if (check(b)) {
					return "B";
				}
			}
			cnt++;
		}
		if (cnt == 10) {
			return "Draw";
		}
		return "Pending";
	}
	bool check(bitset<9> x) {
		for (const int mask : masks) {
			bitset<9> y(mask);
			bitset<9> r = x & y;
			if (r.count() == 3) {
				return true;
			}
		}
		return false;
	}
};
// @lc code=end