题目描述
对于一个给定的井字棋棋盘,请设计一个高效算法判断当前玩家是否获胜。
给定一个二维数组board,代表当前棋盘,其中元素为1的代表是当前玩家的棋子,为0表示没有棋子,为-1代表是对方玩家的棋子。
测试样例:
[[1,0,1],[1,-1,-1],[1,-1,0]]
返回:true
思路
井字棋有四种情况表示当前玩家获胜,1代表当前玩家棋子
- 行全为1, 即行的和为3
- 列全为1, 列的和为3
- 主对角全为1, 对角和为3
- 副对角全为1, 对角和为3
- 如果扩展为N*N的话,判断和是否等于N,下面代码适用任何情况
解答代码:
class Board {
public:
bool checkWon(vector<vector<int> > board) {
int row = board.size();
int i,j,sum;
//检查每一行的和是是否等于row
for(i = 0; i < row; i++){
sum = 0;
for(j = 0; j < row; j++){
sum += board[i][j];
}
if(sum == row) return true;
}
// 检查每一列的和是是否等于row
for(j = 0; j < row; j++){
sum = 0;
for(i = 0; i < row; i++){
sum += board[i][j];
}
if(sum == row) return true;
}
//检查主对角线的和是是否等于row
sum = 0;
for(i = 0; i < row; i++){
sum += board[i][i];
}
if(sum == row) return true;
//检查副对角线的和是是否等于row
sum = 0;
for(i = 0; i < row; i++){
sum += board[i][row - 1 - i];
}
if(sum == row) return true;
// 以上情况都不满足
return false;
}
};
class Board {
public:
bool checkWon(vector<vector<int> > board) {
int len = board.size();
int row = 0, col = 0, s1 = 0, s2 = 0;
for(int i = 0; i < len; ++i){
for(int j = 0; j < len; ++j){
row += board[i][j];
col += board[j][i];
}
if(row == len || col == len) return true;
s1 += board[i][i];
s2 += board[i][len-1];
}
if (s1 == len || s1 == len) return true;
return false;
}
};
如有不同见解,欢迎留言讨论~~