[릿코드] 999번 : Available Captures for Rook

leetcode.com/problems/available-captures-for-rook/

Available Captures for Rook - LeetCode

 

---

 

문제

 

화이트 룩 'R'은 한개, 화이트 비숍 'B'과 블랙 폰 'p' 여러개와 빈공간 '.'로 구성되어있는 8*8 체스보드가 있다.

룩의 이동 ->

동서남북으로 몇칸이든 이동가능.

블랙 폰을 잡거나 화이트비숍에 의해 이동하지 못할 수 있음.

룩 차례에 한번에 잡을 수 있는 폰의 개수는?

 

어떻게 풀까

먼저 룩, 폰, 비숍의 위치를 저장한다.

for문을 폰 크기만큼 돌린다.

이때 룩이 폰의 위치로 이동가능한지 확인

-> 행 혹은 열이 같은지

-> 둘중 하나라도 같으면 -> 행이 같은지 -> 같은 행에 '.' 아닌거 있는지

                                 -> 열이 같은지 -> 같은 열에 '.' 아닌거 있는지

 

구현 중 만난 문제/해결

행이 같을 때와 열이 같을 때만 if (isPossible) result++; 를 통해서 결과에 반영해야했는데 if(){...} else if(){...} if (isPossible) result++; 이런식으로 해버려서 틀렸었다.

마무리가 부족했던것같다ㅠ

 

구현

 

class Solution {
    // 999
    class rowCol {
        int row;
        int col;
        rowCol(int row, int col) {
            this.row = row;
            this.col = col;
        }
    }

   public int numRookCaptures(char[][] board) {
        List<rowCol> pawn = new ArrayList<>();
        List<rowCol> bshop = new ArrayList<>();
        int result = 0;
        rowCol rook = new rowCol(0,0);
        // 폰, 비숍, 룩의 위치 등록
        for (int i=0; i<board.length; i++) {
            for (int j=0; j<board.length; j++) {
                if (board[i][j] == '.') {
                    continue;
                } else if (board[i][j] == 'B') {
                    bshop.add(new rowCol(i,j));
                } else if(board[i][j] == 'p') {
                    pawn.add(new rowCol(i,j));
                } else if(board[i][j] == 'R') {
                    rook = new rowCol(i,j);
                }
            }
        }

        int rookRow = rook.row;
        int rookCol = rook.col;
        for (int i=0; i<pawn.size(); i++) {
            rowCol curPawn = pawn.get(i);
            int row = curPawn.row;
            int col = curPawn.col;
            boolean isPossible = true;
            if (row == rookRow) {
            	// 행이 같은 경우 룩과 폰사이에 비숍만없으면 잡기 가능
                char[] boardI = board[row];
                int min = col;
                int max = rookCol;
                if (max < min) {
                    min = rookCol;
                    max = col;
                }
                for (int j = min+1; j<max; j++) {
                    if (boardI[j] != '.') isPossible = false;
                }
                if (isPossible) result++;
            } else if (col == rookCol) {
                // 열이 같은 경우 룩과 폰사이에 비숍만없으면 잡기 가능
                int min = row;
                int max = rookRow;
                if (max < min) {
                    min = rookRow;
                    max = row;
                }
                for (int j = min+1; j<max; j++) {
                    if (board[j][col] != '.') isPossible = false;
                }
                if (isPossible) result++;
            }
        }

        return result;
    }
}