设计井字棋

/*
 * 设计井字棋
 * 题目思路:
 * 实现一个TicTacToe类，它具备如下功能:
 * 1. TicTacToe(int n) 初始化一个大小为 n 的井字棋棋盘。
 * 2. int move(int row, int col, int player) 表示玩家 player 在棋盘 (row, col) 上落子。
 *    - player = 1 表示玩家 1，player = 2 表示玩家 2。
 *    - 如果玩家 player 获胜，返回 player。如果没有玩家获胜，返回 0。
 */
 
import java.util.stream.IntStream;
 
public class TicTacToe {
    private int[] rows;
    private int[] cols;
    private int diagonal;
    private int antiDiagonal;
    private int n;
 
    public TicTacToe(int n) {
        this.n = n;
        rows = new int[n];
        cols = new int[n];
        diagonal = 0;
        antiDiagonal = 0;
    }
 
    public int move(int row, int col, int player) {
        int toAdd = player == 1 ? 1 : -1;
 
        rows[row] += toAdd;
        cols[col] += toAdd;
 
        if (row == col) {
            diagonal += toAdd;
        }
 
        if (row + col == n - 1) {
            antiDiagonal += toAdd;
        }
 
        if (IntStream.of(rows).anyMatch(x -> Math.abs(x) == n) ||
            IntStream.of(cols).anyMatch(x -> Math.abs(x) == n) ||
            Math.abs(diagonal) == n || Math.abs(antiDiagonal) == n) {
            return player;
        }
 
        return 0;
    }
}