본문 바로가기
백준알고리즘/구현

(Python/🥈3)백준 알고리즘 1063번: 킹

by windy7271 2023. 12. 28.
728x90
반응형

 

문제 바로가기 

킹

 

문제:

8*8크기의 체스판에 왕이 하나 있다. 킹의 현재 위치가 주어진다. 체스판에서 말의 위치는 다음과 같이 주어진다. 알파벳 하나와 숫자 하나로 이루어져 있는데, 알파벳은 열을 상징하고, 숫자는 행을 상징한다. 열은 가장 왼쪽 열이 A이고, 가장 오른쪽 열이 H까지 이고, 행은 가장 아래가 1이고 가장 위가 8이다. 예를 들어, 왼쪽 아래 코너는 A1이고, 그 오른쪽 칸은 B1이다. 킹은 다음과 같이 움직일 수 있다.

 

 

R : 한 칸 오른쪽으로

L : 한 칸 왼쪽으로

B : 한 칸 아래로

T : 한 칸 위로

RT : 오른쪽 위 대각선으로

LT : 왼쪽 위 대각선으로

RB : 오른쪽 아래 대각선으로

LB : 왼쪽 아래 대각선으로 체스판에는 돌이 하나 있는데, 돌과 같은 곳으로 이동할 때는, 돌을 킹이 움직인 방향과 같은 방향으로 한 칸 이동시킨다.

아래 그림을 참고하자.

입력으로 킹이 어떻게 움직여야 하는지 주어진다. 입력으로 주어진 대로 움직여서 킹이나 돌이 체스판 밖으로 나갈 경우에는 그 이동은 건너 뛰고 다음 이동을 한다. 킹과 돌의 마지막 위치를 구하는 프로그램을 작성하시오.

입력:

첫째 줄에 킹의 위치, 돌의 위치, 움직이는 횟수 N이 주어진다. 둘째 줄부터 N개의 줄에는 킹이 어떻게 움직여야 하는지 주어진다. N은 50보다 작거나 같은 자연수이고, 움직이는 정보는 위에 쓰여 있는 8가지 중 하나이다.

 

출력:

첫째 줄에 킹의 마지막 위치, 둘째 줄에 돌의 마지막 위치를 출력한다.

 

풀이:

1. 파이썬

import sys

move = {'R': [1, 0], 'L': [-1, 0], 'B': [0, -1], 'T': [0, 1], 'RT': [1, 1], 'LT': [-1, 1], 'RB': [1, -1], 'LB': [-1, -1]}
number = {"A":1,
          "B":2,
          "C":3,
          "D":4,
          "E":5,
          "F":6,
          "G":7,
          "H":8}
target, start, n = list(input().split(" "))
king = [number[target[0]], int(target[1])]
stone = [number[start[0]], int(start[1])]

for i in range(int(n)):
    mo = input()
    nx = king[0] + move[mo][0]; ny = king[1] + move[mo][1]
    if 0 < nx <= 8 and 0 < ny <= 8 :
        # 킹이 벽돌로 가고,
        if nx == stone[0] and ny == stone[1]:
            # 킹이 가는 방향으로 간다.
            sx = stone[0] + move[mo][0]
            sy = stone[1] + move[mo][1]
            if 0<sx<=8 and 0<sy<=8:
                king = [nx,ny]
                stone = [sx,sy]
        else:
            king = [nx,ny]
for i in king, stone:
    for key, value in number.items():
        if i[0] == value:
            print(key+str(i[1]))

 

 

이동경로와, 알파벳별 위치를 딕셔너리로 적었다.

 

마지막 for문에서 두개의 리스트를 하나 안에 넣어도 작동 가능하다.

 

2.자바

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Map;

public class Main {
    static int[] move(String s) {
        int x = 0;
        int y = 0;
        if (s.equals("R")) {
            x = 1;
        } else if (s.equals("L")) {
            x = -1;
        } else if (s.equals("B")) {
            y = -1;
        } else if (s.equals("T")) {
            y = 1;
        } else if (s.equals("RT")) {
            x = 1;
            y = 1;
        } else if (s.equals("LT")) {
            x = -1;
            y = 1;
        } else if (s.equals("RB")) {
            x = 1;
            y = -1;
        } else if (s.equals("LB")) {
            x = -1;
            y = -1;
        }
        return new int[]{x, y};
    }

    static int[] king, stone;
    static int n,nx,ny;
    static Map<String, Integer> map = Map.of(
            "A", 1,
            "B", 2,
            "C", 3,
            "D", 4,
            "E", 5,
            "F", 6,
            "G", 7,
            "H", 8
    );

    public static void main(String[] args) throws IOException {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        String[] s = bf.readLine().split(" ");
        int n = Integer.parseInt(s[2]);
//        king ,stone 해쉬로 값 가져오기
        int[] king = new int[]{ map.get(String.valueOf(s[0].charAt(0))), s[0].charAt(1) - '0'}; // [1,1]
        int[] stone = new int[]{map.get(String.valueOf(s[1].charAt(0))), s[1].charAt(1) - '0'}; // [1,2]

        for (int i = 0; i < n; i++) {

            int[] mo = move(bf.readLine());
            int nx = king[0] + mo[0];
            int ny = king[1] + mo[1];
            if (0 < nx && nx <= 8 && 0 < ny && ny <= 8) {
                if (nx == stone[0] && ny == stone[1]) {
                    int sx = stone[0] + mo[0];
                    int sy = stone[1] + mo[1];
                    if (0 < sx && sx <= 8 && 0 < sy && sy <= 8) {
                        king = new int[]{nx, ny};
                        stone = new int[]{sx, sy};
                    }
                }
                else {
                    king = new int[]{nx, ny};
                }
            }
        }
        System.out.println((char)(64+king[0])+""+(king[1]));
        System.out.println((char)(64+stone[0])+""+(stone[1]));
    }
}

 

 

move 클래스를 따로 만들어 입력값에 따라 리턴을 해주는 식으로 했다.

 

반응형

댓글