999.可以被一步捕获的棋子数

125 阅读1分钟

题目:
在一个 8 x 8 的棋盘上,有一个白色的车(Rook),用字符 'R' 表示。棋盘上还可能存在空方块,白色的象(Bishop)以及黑色的卒(pawn),分别用字符 '.''B' 和 'p' 表示。不难看出,大写字符表示的是白棋,小写字符表示的是黑棋。

车按国际象棋中的规则移动。东,西,南,北四个基本方向任选其一,然后一直向选定的方向移动,直到满足下列四个条件之一:

  • 棋手选择主动停下来。
  • 棋子因到达棋盘的边缘而停下。
  • 棋子移动到某一方格来捕获位于该方格上敌方(黑色)的卒,停在该方格内。
  • 车不能进入/越过已经放有其他友方棋子(白色的象)的方格,停在友方棋子前。

你现在可以控制车移动一次,请你统计有多少敌方的卒处于你的捕获范围内(即,可以被一步捕获的棋子数)。 算法:
方法一:遍历

func numRookCaptures(board [][]byte) int {
	count := 0
	row, col := len(board), len(board[0])
	for i := range board {
		for j := range board[i] {
			if board[i][j] == byte('R') {
				for left := j - 1; left >= 0 && board[i][left] != byte('B'); left -- {
					if board[i][left] == byte('p') {
						count ++
						break
					}
				}
				for right := j + 1; right < col && board[i][right] != byte('B'); right ++ {
					if board[i][right] == byte('p') {
						count ++
						break
					}
				}
				for up := i - 1; up >= 0 && board[up][j] != byte('B'); up -- {
					if board[up][j] == byte('p') {
						count ++
						break
					}
				}
				for down := i + 1; down < row  && board[down][j] != byte('B'); down ++ {
					if board[down][j] == byte('p') {
						count ++
						break
					}
				}
				return count
			}
		}
	}
	return 0
}