[SWEA] 1211. [S/W 문제해결 기본] Array2 - Ladder2

문제

https://www.swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV14BgD6AEECFAYh&categoryId=AV14BgD6AEECFAYh&categoryType=CODE

SW Expert Academy

 

 

풀이

이 문제는 Ladder1 문제를 약간 응용한 문제이다. 모든 시작점을 기준으로 바닥에 도달하는 경로 중 가장 짧은 코스의 좌표를 구하는 문제이다. 내가 한 방식은 처음에 사디리 정보를 받을 때 시작점을 저장하는 벡터를 이용해서 저장했다. 하지만 10개의 케이스를 할 때마다 벡터를 새로 만들거나 초기화 해줬어야 하는데 이 부분을 간과해서 몇시간 동안 엉뚱한 답으로 헤맸었다. 다음부턴 초기화 부분에 더 신경을 써야겠다.

 

 

import java.util.*;
 
class Point{
    int row,col;
}
public class Solution {
    static Scanner scan = new Scanner(System.in);
    static int[][] map = new int[100][100];
    static int flag = 0; // 0 : 아래, 1 : 왼쪽, 2 : 오른쪽
    static Vector<Integer> vec;
            
    static void read() {
        scan.nextInt();
        vec = new Vector<Integer>();
        for (int i = 0; i < 100; i++) {
            for (int j = 0; j < 100; j++) {
                map[i][j] = scan.nextInt();
                if (i==0 &&map[i][j] == 1)
                    vec.add(j);
            }
        }
    }
 
    static int get_num() {
        int idx = -1, dis_min = Integer.MAX_VALUE;
        for(int i=0;i<vec.size();i++) {
            int r = 0, c = vec.get(i), cnt=1;
            flag = 0;
            while (true) {
                if (r == 99) {
                    if(cnt<=dis_min) {
                        idx = vec.get(i);
                        dis_min = cnt;
                    } 
                    break;
                }
                if (flag == 0) {// 아래 방향
                    if (is_valid(r, c - 1)) {// 왼쪽에 존재
                        c--;
                        flag = 1;
                    } else if (is_valid(r, c + 1)) {// 오른쪽에 존재
                        c++;
                        flag = 2;
                    } else // 양쪽에 없음
                        r++;
                } else {// 꺾인 상태
                    if (flag == 1) {// 왼쪽
                        if (is_valid(r + 1, c)) {
                            r++;
                            flag = 0;
                        } else
                            c--;
 
                    } else {// 오른쪽
                        if (is_valid(r + 1, c)) {
                            r++;
                            flag = 0;
                        } else
                            c++;
                    }
                }
                cnt++;
            }
        }
        return idx;
    }
 
    static boolean is_valid(int r, int c) {
        if ((r >= 0 && r < 100) && (c >= 0 && c < 100) && (map[r][c] != 0))
            return true;
        return false;
    }
 
    static void solution() {
        for (int i = 1; i <= 10; i++) {
            read();
            System.out.println("#" + i + " " + get_num());
        }
    }
 
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        solution();
    }
 
}