[SWEA] 5650. [모의 SW 역량테스트] 핀볼 게임 Java

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWXRF8s6ezEDFAUo 

SW Expert Academy

---

1. 문제 요약

NxN크기의 map에서 핀볼이 이동하는데,

빈칸(0)을 만날 경우, 방향 그대로 유지한다.

웜홀(6~10)을 만날 경우, 같은 웜홀의 다른 좌표로 현재 위치가 바뀐다.

블록(1~5)을 만날 경우, 지금 방향으로 갔을 때 블록의 수평면일 경우 - 방향이 반대가 되고,

                                     지금 방향으로 갔을 때 블록의 경사면일 경우 - 방향이 직각 방향으로 바뀐다.(반사되는 방향)

시작지점 또는 블랙홀(-1)을 만날 경우, 이동을 멈추고 점수의 최대값을 갱신한다.

주어진 map에서 얻을 수 있는 최대 점수를 출력한다.

 

* 시작지점은 빈 칸일 때만 가능하다.

 

2. 풀이

시뮬레이션 문제이다.

입력받을 때, map의 값이 6 이상일 경우 웜홀 배열에 좌표를 저장하였다.

웜홀 배열은 warm = new Pos[11][2] 로 설정했는데, 

map의 숫자를 웜홀의 인덱스로 사용하기 위해 11,

같은 인덱스로 웜홀이 2개가 존재하므로 2 로 설정하고 시작했다.

그리고 y, x 좌표를 저장할 것이므로 Pos 클래스로 배열을 생성하였다.

 

map에서 모든 빈칸(0)을 시작지점으로 설정해보고 얻을 수 있는 점수의 최댓값을 갱신하는 방식이다.

1. 벽에 부딪힐 때 (가장 맨 위에 와야함 -> 좌표 범위 때문)

2. 시작점, 블랙홀일 때

3. 웜홀일 때

4. 빈칸일 때

5. 블록에 부딪힐 때 

조건을 나눠서 구현하였다.

 

내가 놓친 부분은

1. 자꾸 무한루프가 돌았다. 원인은 2가지였다.

  - 빈칸일 때를 생각해주지 않았다. map 값이 0일 때에는 아무 일도 수행하지 않으니 계속 무한루프가 돌았다.

  - 블록에 부딪힐 때 방향 바꿔주는 부분에서 방향을 하나 잘못 바꿔주고 있었다. => 이게 제일 까다로웠다.. 제일 오래걸렸음

 

(+)

풀이1

블록에 부딪힐 때, 블록마다 if문으로 처리를 해줬었다.

if (map[ny][nx] >= 1 && map[ny][nx] <= 5) {
                int idx = map[ny][nx];
                if (idx == 1) {
                    //수평면
                    if (d == 0) d = 1;
                    else if (d == 3) d = 2;
                    // 경사면
                    else if (d == 1) d = 3;
                    else if (d == 2) d = 0;
                }
                
                else if (idx == 2) {
                	//수평면
                    if (d == 1) d = 0;
                    else if (d == 3) d = 2;
                    // 경사면
                    else if (d == 0) d = 3;
                    else if (d == 2) d = 1;
                }
                
                else if (idx == 3) {
                    //수평면
                    if (d == 1) d = 0;
                    else if (d == 2) d = 3;
                    // 경사면
                    else if (d == 0) d = 2;
                    else if (d == 3) d = 1;
                }
                
                else if (idx == 4) {
                    //수평면
                    if (d == 0) d = 1;
                    else if (d == 2) d = 3;
                    // 경사면
                    else if (d == 1) d = 2;
                    else if (d == 3) d = 0;
                }
                
                else if (idx == 5) {
					//수평면
                    if (d == 0) d = 1;
                    else if (d == 1) d = 0;
                    else if (d == 2) d = 3;
                    else if (d == 3) d = 2;
				}
                score++;
				continue;
            }

풀이2

다른 방법으로도 풀어봤는데, 탈주범 검거 풀 때 파이프 연결 방향 생각해줄 때처럼 풀어봤다.

핀볼이 오른쪽으로 가고있었으면, 블록의 왼쪽에 부딪힌다. 즉, 현재 핀볼 방향의 반대 방향이 블록에 부딪히는 방향이다. 

이 점을 이용해서 블록의 방향마다 수평면이면 1, 경사면이면 0 을 저장해주었다.

수평면을 만나면 핀볼의 방향을 반대로 바꿔주고,

경사면을 만나면 블록의 경사면인 또 다른 방향이 핀볼의 방향이 된다.

// 블록에 부딪힐 때
            if (map[ny][nx] >= 1 && map[ny][nx] <= 5) {
                int idx = map[ny][nx];
                
                //부딪히는 블록의 방향
                int bd = 0;
                switch(d) { //현재방향의 반대방향 == 부딪히는 블록의 방향
	                case 0: bd=1; break;
	                case 1: bd=0; break;
	                case 2: bd=3; break;
	                case 3: bd=2; break;
                }
                
                if ( block[idx][bd]==1 ) { //수평이면 반대방향
                	d = bd;
                } else if ( block[idx][bd]==0 ) { //수직이면 직각방향
                	for (int i=0; i<4; i++) {
                		if ( bd!=i && block[idx][i]==0 ) {
                			d=i;
                    		break;
                		}
                	}
                }
                score++;
				continue;
            }

 

 

3. 코드

블록에 부딪힐 때 - if문

package a22_10_20;

import java.io.*;
import java.util.*;

public class SW_5650_핀볼게임 {

    static int[][] map;
    static int N, ans;
    static Pos[][] warm;

    static int[] dy = { -1, 1, 0, 0 };
    static int[] dx = { 0, 0, -1, 1 };

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;
        int T = Integer.parseInt(br.readLine());
        for (int tc = 1; tc <= T; tc++) {
            N = Integer.parseInt(br.readLine());
            map = new int[N][N];
            warm = new Pos[11][2];
            
            for (int i = 0; i < N; i++) {
                st = new StringTokenizer(br.readLine());
                for (int j = 0; j < N; j++) {
                	int num = Integer.parseInt(st.nextToken());
                    map[i][j] = num;
                    if (num >= 6) {
                    	if (warm[ num ][0] == null) {
                    		warm[num][0] = new Pos(i, j);
                    	} else {
                    		warm[num][1] = new Pos(i, j);
                    	}
                    }
                }
            } // 입력 끝

            ans = 0;
            solution();
            System.out.println("#" + tc + " " + ans);
        }
    }

    static void solution() {

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (map[i][j] != 0) continue;
                for (int d = 0; d < 4; d++) {
                	move(i, j, d);
                }
            }
        }
    }

    static void move(int sy, int sx, int d) {
        int score = 0;
        int ny = sy;
        int nx = sx;
        while (true) {
            ny += dy[d];
            nx += dx[d];

            // 벽에 부딪힐 때
            if (ny < 0 || nx < 0 || ny >= N || nx >= N) {
            	if (d == 0) d = 1;
                else if (d == 1) d = 0;
                else if (d == 2) d = 3;
                else if (d == 3) d = 2;
                score++;
                continue;
            }
            // 시작점이거나, 블랙홀일 때
            if ((ny == sy && nx == sx) || map[ny][nx] == -1) {
                ans = Math.max(score, ans);
                break;
            }
            // 웜홀일 때
            if (map[ny][nx] >= 6) {
                int idx = map[ny][nx];
                if (warm[idx][0].y == ny && warm[idx][0].x == nx) {
                    ny = warm[idx][1].y;
                    nx = warm[idx][1].x;
                } else if (warm[idx][1].y == ny && warm[idx][1].x == nx){
                    ny = warm[idx][0].y;
                    nx = warm[idx][0].x;
                }
                continue;
            }
            //빈칸이면 쭉 간다
            if (map[ny][nx] == 0) continue;
            // 블록에 부딪힐 때
            if (map[ny][nx] >= 1 && map[ny][nx] <= 5) {
                int idx = map[ny][nx];
                if (idx == 1) {
                    //수평면
                    if (d == 0) d = 1;
                    else if (d == 3) d = 2;
                    // 경사면
                    else if (d == 1) d = 3;
                    else if (d == 2) d = 0;
                }
                
                else if (idx == 2) {
                	//수평면
                    if (d == 1) d = 0;
                    else if (d == 3) d = 2;
                    // 경사면
                    else if (d == 0) d = 3;
                    else if (d == 2) d = 1;
                }
                
                else if (idx == 3) {
                    //수평면
                    if (d == 1) d = 0;
                    else if (d == 2) d = 3;
                    // 경사면
                    else if (d == 0) d = 2;
                    else if (d == 3) d = 1;
                }
                
                else if (idx == 4) {
                    //수평면
                    if (d == 0) d = 1;
                    else if (d == 2) d = 3;
                    // 경사면
                    else if (d == 1) d = 2;
                    else if (d == 3) d = 0;
                }
                
                else if (idx == 5) {
					//수평면
                    if (d == 0) d = 1;
                    else if (d == 1) d = 0;
                    else if (d == 2) d = 3;
                    else if (d == 3) d = 2;
				}
                score++;
				continue;
            }
        }
    }

    static class Pos {
        int y, x;

        Pos(int y, int x) {
            this.y = y;
            this.x = x;
        }
    }
}

블록에 부딪힐 때 - 블록 2차원 배열

package a22_10_22;

import java.io.*;
import java.util.*;

public class SW_5650_핀볼게임 {

    static int[][] map;
    static int N, ans;
    static Pos[][] warm;

    static int[] dy = { -1, 1, 0, 0 };
    static int[] dx = { 0, 0, -1, 1 };
    static int[][] block = {
    	  // 상 하 좌 우 - 수평:1, 직각:0
    		{0,0,0,0}, //0 dummy
    		{0,1,1,0}, //1
    		{1,0,1,0}, //2
    		{1,0,0,1}, //3
    		{0,1,0,1}, //4
    		{1,1,1,1}  //5
    	};

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;
        int T = Integer.parseInt(br.readLine());
        for (int tc = 1; tc <= T; tc++) {
            N = Integer.parseInt(br.readLine());
            map = new int[N][N];
            warm = new Pos[11][2];
            
            for (int i = 0; i < N; i++) {
                st = new StringTokenizer(br.readLine());
                for (int j = 0; j < N; j++) {
                	int num = Integer.parseInt(st.nextToken());
                    map[i][j] = num;
                    if (num >= 6) {
                    	if (warm[ num ][0] == null) {
                    		warm[num][0] = new Pos(i, j);
                    	} else {
                    		warm[num][1] = new Pos(i, j);
                    	}
                    }
                }
            } // 입력 끝

            ans = 0;
            solution();
            System.out.println("#" + tc + " " + ans);
        }
    }

    static void solution() {

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (map[i][j] != 0) continue;
                for (int d = 0; d < 4; d++) {
                	move(i, j, d);
                }
            }
        }
    }

    static void move(int sy, int sx, int d) {
        int score = 0;
        int ny = sy;
        int nx = sx;
        while (true) {
            ny += dy[d];
            nx += dx[d];

            // 벽에 부딪힐 때
            if (ny < 0 || nx < 0 || ny >= N || nx >= N) {
            	if (d == 0) d = 1;
                else if (d == 1) d = 0;
                else if (d == 2) d = 3;
                else if (d == 3) d = 2;
                score++;
                continue;
            }
            // 시작점이거나, 블랙홀일 때
            if ((ny == sy && nx == sx) || map[ny][nx] == -1) {
                ans = Math.max(score, ans);
                break;
            }
            // 웜홀일 때
            if (map[ny][nx] >= 6) {
                int idx = map[ny][nx];
                if (warm[idx][0].y == ny && warm[idx][0].x == nx) {
                    ny = warm[idx][1].y;
                    nx = warm[idx][1].x;
                } else if (warm[idx][1].y == ny && warm[idx][1].x == nx){
                    ny = warm[idx][0].y;
                    nx = warm[idx][0].x;
                }
                continue;
            }
            
            //빈칸이면 쭉 간다
            if (map[ny][nx] == 0) continue;
            
            // 블록에 부딪힐 때
            if (map[ny][nx] >= 1 && map[ny][nx] <= 5) {
                int idx = map[ny][nx];
                
                //부딪히는 블록의 방향
                int bd = 0;
                switch(d) { //현재방향의 반대방향 == 부딪히는 블록의 방향
	                case 0: bd=1; break;
	                case 1: bd=0; break;
	                case 2: bd=3; break;
	                case 3: bd=2; break;
                }
                
                if ( block[idx][bd]==1 ) { //수평이면 반대방향
                	d = bd;
                } else if ( block[idx][bd]==0 ) { //수직이면 직각방향
                	for (int i=0; i<4; i++) {
                		if ( bd!=i && block[idx][i]==0 ) {
                			d=i;
                    		break;
                		}
                	}
                }
                score++;
				continue;
            }
        }
    }

    static class Pos {
        int y, x;

        Pos(int y, int x) {
            this.y = y;
            this.x = x;
        }
    }
}