Yoo Chaewon

Yoo Chaewon

신나고 재밌게🤪

[ALGORITHM]BOJ14890-경사로

문제

크기가 N×N인 지도가 있다. 지도의 각 칸에는 그 곳의 높이가 적혀져 있다.

오늘은 이 지도에서 지나갈 수 있는 길이 몇 개 있는지 알아보려고 한다. 길이란 한 행 또는 한 열 전부를 나타내며, 한쪽 끝에서 다른쪽 끝까지 지나가는 것이다.

다음과 같은 N=6인 경우 지도를 살펴보자.

img

이때, 길은 총 2N개가 있으며, 아래와 같다.

img

길을 지나갈 수 있으려면 길에 속한 모든 칸의 높이가 모두 같아야 한다. 또는, 경사로를 놓아서 지나갈 수 있는 길을 만들 수 있다. 경사로는 높이가 항상 1이며, 길이는 L이다. 또, 개수는 매우 많아 부족할 일이 없다. 경사로는 낮은 칸과 높은 칸을 연결하며, 아래와 같은 조건을 만족해야한다.

  • 경사로는 낮은 칸에 놓으며, L개의 연속된 칸에 경사로의 바닥이 모두 접해야 한다.
  • 낮은 칸과 높은 칸의 높이 차이는 1이어야 한다.
  • 경사로를 놓을 낮은 칸의 높이는 모두 같아야 하고, L개의 칸이 연속되어 있어야 한다.

아래와 같은 경우에는 경사로를 놓을 수 없다.

  • 경사로를 놓은 곳에 또 경사로를 놓는 경우
  • 낮은 칸과 높은 칸의 높이 차이가 1이 아닌 경우
  • 낮은 지점의 칸의 높이가 모두 같지 않거나, L개가 연속되지 않은 경우
  • 경사로를 놓다가 범위를 벗어나는 경우

L = 2인 경우에 경사로를 놓을 수 있는 경우를 그림으로 나타내면 아래와 같다.

img

경사로를 놓을 수 없는 경우는 아래와 같다.

img

위의 그림의 가장 왼쪽부터 1번, 2번, 3번, 4번 예제라고 했을 때, 1번은 높이 차이가 1이 아니라서, 2번은 경사로를 바닥과 접하게 놓지 않아서, 3번은 겹쳐서 놓아서, 4번은 기울이게 놓아서 불가능한 경우이다.

가장 위에 주어진 그림 예의 경우에 지나갈 수 있는 길은 초록색으로, 지나갈 수 없는 길은 빨간색으로 표시되어 있으며, 아래와 같다. 경사로의 길이 L = 2이다.

img

지도가 주어졌을 때, 지나갈 수 있는 길의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (2 ≤ N ≤ 100)과 L (1 ≤ L ≤ N)이 주어진다. 둘째 줄부터 N개의 줄에 지도가 주어진다. 각 칸의 높이는 10보다 작거나 같은 자연수이다.

출력

첫째 줄에 지나갈 수 있는 길의 개수를 출력한다.

예제 입력 1

6 2
3 3 3 3 3 3
2 3 3 3 3 3
2 2 2 3 2 3
1 1 1 2 2 2
1 1 1 3 3 1
1 1 2 3 3 2

예제 출력 1

3

예제 입력 2

6 2
3 2 1 1 2 3
3 2 2 1 2 3
3 2 2 2 3 3
3 3 3 3 3 3
3 3 3 3 2 2
3 3 3 3 2 2

예제 출력 2

7

예제 입력 3

6 3
3 2 1 1 2 3
3 2 2 1 2 3
3 2 2 2 3 3
3 3 3 3 3 3
3 3 3 3 2 2
3 3 3 3 2 2

예제 출력 3

3

예제 입력 4

6 1
3 2 1 1 2 3
3 2 2 1 2 3
3 2 2 2 3 3
3 3 3 3 3 3
3 3 3 3 2 2
3 3 3 3 2 2

예제 출력 4

11

힌트

예제 2의 경우 아래와 같은 초록색 길을 지나갈 수 있다.

img

예제 3의 경우에는 아래와 같은 초록색 길이 지나갈 수 있는 길이다.

img

마지막으로, 예제 4의 경우에는 아래와 같은 초록색 길이 지나갈 수 있는 길이다.

img

나의 코드 ver.1

import java.io.InputStreamReader;
import java.util.Scanner;

public class Main {
    static int[][] map;
    static boolean[][] visited1;
    static boolean[][] visited2;
    static int N;// 지도 길이
    static int L;// 경사로 길이

    public static void main(String[] args) {
        Scanner scanner = new Scanner(new InputStreamReader(System.in));

        N = scanner.nextInt();
        L = scanner.nextInt();
        map = new int[N][N];
        visited1 = new boolean[N][N];
        visited2 = new boolean[N][N];

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                map[i][j] = scanner.nextInt();
            }
        }

        int answer = 0;
        //가로 검사
        for (int i = 0; i < N; i++) {
            int pre = map[i][0];
            answer++;
            for (int j = 1; j < N; j++) {
                if (pre != map[i][j]) {
                    if (pre - map[i][j] < 0) {//오르막
                        if (!upper(map[i][j], j, i, 0)) {
                            answer--;
                            break;
                        }
                    } else {//내리막
                        if (!lower(map[i][j - 1], j-1, i, 0)) {
                            answer--;
                            break;
                        } else {//성공
                            j += L - 1;
                        }
                    }
                }
                pre = map[i][j];
            }
        }

        //세로 검사
        for (int j = 0; j < N; j++) {
            int pre = map[0][j];
            answer++;
            for (int i = 1; i < N; i++) {
                if (pre != map[i][j]) {
                    if (pre - map[i][j] < 0) {//오르막
                        if (!upper(map[i][j], j, i, 1)) {
                            answer--;
                            break;
                        }
                    } else {//내리막
                        if (!lower(map[i - 1][j], j, i - 1, 1)) {
                            answer--;
                            break;
                        } else {//성공
                            i += L - 1;
                        }
                    }
                }
                pre = map[i][j];
            }
        }
        System.out.print(answer);
    }

    //dir:0 - 가로방향, dir:1 - 세로방향
    public static boolean upper(int cur, int x, int y, int dir) {//오르막 경사
        if (dir == 0) {
            if (x - L < 0) return false;
            for (int i = 1; i <= L; i++) {
                if (map[y][x - i] != cur - 1 || visited1[y][x - i]) return false;
                visited1[y][x - i] = true;
            }
        } else {
            if (y - L < 0) return false;
            for (int i = 1; i <= L; i++) {
                if (map[y - i][x] != cur - 1 || visited2[y - i][x]) return false;
                visited2[y - i][x] = true;
            }
        }
        return true;
    }

    public static boolean lower(int cur, int x, int y, int dir) {//내리막 경사로
        if (dir == 0) {
            if (x + L >= N) return false;
            for (int i = 1; i <= L; i++) {
                if (map[y][x + i] != cur - 1 || visited1[y][x + i]) return false;
                visited1[y][x + i] = true;
            }
        } else {
            if (y + L >= N) return false;
            for (int i = 1; i <= L; i++) {
                if (map[y + i][x] != cur - 1 || visited2[y + i][x]) return false;
                visited2[y + i][x] = true;
            }
        }
        return true;
    }
}
  • 먼저 나는 크게 가로검사/ 세로검사로 나누었다.
  • 다시 가로(세로)인 경우 👉 오르막 경사/ 내리막 경사인 경우를 또 나누었다.
    • 오르막인지 내리막인지는 if (pre - map[ i ][ j ] < 0) 으로, 이전보다 현재 값이 크면 오르막, 반대는 내리막을 실행시켰다.
  • 항상 answer++을 해주었고, 만약에 오르막, 내리막을 실패할 경우는 -1해서 답을 얻었다.
  • upper 와 lower함수는 경사로를 놓을 수 있는지 없는지를 판별해주는 함수다. 이를 세로인 경우, 가로인 경우를 나눠서 작성했다. 나누지 않고 할 수 있는지 고민해봐야겠다.
  • visited를 사용하지 않았었는데, 332233(L=2)인 경우 true가 나왔다. 22인 부분을 경사로 세워준것을 표시하기 위해 visited를 사용하였다. 이를 visited 2개 사용한 점이 아쉽다.
  • 까탈스러웠던 부분은 index로 하는거다 보니 많이 어려웠다.

느낀점

빨리 풀고 싶어서 생각을 많이 하지 않고 바로 코드 부터 쳤다. 그러다 보니 작성하면서도 비슷한 것들이 많아서 좀만 고민을 더 해보면 짧게 할 수 있겠다? 라는 생각이 들었다. 아무튼 오늘은 풀었으니 좀 더 바꾸는 것은 다음에 할 것이다‼️첫 삼성 문제였는데 역시 길다! 나만 길수도! 아무튼 풀고 나니 뿌듯하다.

  • 문제: https://www.acmicpc.net/problem/14890
  • 저장소: https://github.com/yoo-chaewon/HELLO_JAVA/blob/master/Algorithm/1.BOJ/SAMSUNG_SW/Q_14890_경사로.java

참고