백준 3109: 빵집 - 그리디, 그래프 탐색(파이썬, 자바)

빵집 성공다국어

3109번: 빵집

문제

유명한 제빵사 김원웅은 빵집을 운영하고 있다. 원웅이의 빵집은 글로벌 재정 위기를 피해가지 못했고, 결국 심각한 재정 위기에 빠졌다.

원웅이는 지출을 줄이고자 여기저기 지출을 살펴보던 중에, 가스비가 제일 크다는 것을 알게되었다. 따라서 원웅이는 근처 빵집의 가스관에 몰래 파이프를 설치해 훔쳐서 사용하기로 했다.

빵집이 있는 곳은 R*C 격자로 표현할 수 있다. 첫째 열은 근처 빵집의 가스관이고, 마지막 열은 원웅이의 빵집이다.

원웅이는 가스관과 빵집을 연결하는 파이프를 설치하려고 한다. 빵집과 가스관 사이에는 건물이 있을 수도 있다. 건물이 있는 경우에는 파이프를 놓을 수 없다.

가스관과 빵집을 연결하는 모든 파이프라인은 첫째 열에서 시작해야 하고, 마지막 열에서 끝나야 한다. 각 칸은 오른쪽, 오른쪽 위 대각선, 오른쪽 아래 대각선으로 연결할 수 있고, 각 칸의 중심끼리 연결하는 것이다.

원웅이는 가스를 되도록 많이 훔치려고 한다. 따라서, 가스관과 빵집을 연결하는 파이프라인을 여러 개 설치할 것이다. 이 경로는 겹칠 수 없고, 서로 접할 수도 없다. 즉, 각 칸을 지나는 파이프는 하나이어야 한다.

원웅이 빵집의 모습이 주어졌을 때, 원웅이가 설치할 수 있는 가스관과 빵집을 연결하는 파이프라인의 최대 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 R과 C가 주어진다. (1 ≤ R ≤ 10,000, 5 ≤ C ≤ 500)

다음 R개 줄에는 빵집 근처의 모습이 주어진다. '.'는 빈 칸이고, 'x'는 건물이다. 처음과 마지막 열은 항상 비어있다.

출력

첫째 줄에 원웅이가 놓을 수 있는 파이프라인의 최대 개수를 출력한다.

 

내 코드

import sys

def dfs(x,y):
    global answer

    arr[x][y]='o'

    if y==c-1: #끝까지 도달 했으면,
        answer+=1 #정답 개수 추가
        return True #도달했다고 알림
    

    for i in range(3):
        nx=x+dx[i]
        ny=y+dy[i]

        if nx<0 or nx>=r or ny<0 or ny>=c:
            continue

        if arr[nx][ny]=='.':
            if dfs(nx,ny): #이 곳에서 시작된 것이 끝까지 도달했으면,
                return True #해당 (x,0)에서 시작된 경로 고정 후 탈출.
    
    return False


#메인
r,c=map(int,sys.stdin.readline().split())
arr=[]
for _ in range(r):
    arr.append(list(sys.stdin.readline().rstrip()))

dx=[-1,0,1]
dy=[1,1,1]


answer=0
for i in range(r):
    dfs(i,0)

print(answer)

 

풀이 및 접근)

- 처음에는 모든 경우를 다 돌아보고 언제 최대 개수인지를 세어주려 했다. 그러려 했는데, 엄청나게 많이 재귀를 하게 되었고, 탈출 조건이나 다음 줄로 넘어가는 부분이 잘 되지 않았다.

 

- 해당 문제는 그리디적 사고를 필요로 했다. 윗줄부터 시작했다면, 최대한 위에서 끝까지 도달해야 최대의 개수를 얻을 수 있다. 따라서, 탐색의 우선순위를 대각선 위, 오른쪽, 대각선 아래 순으로 탐색해야 한다. 이 순서대로 탐색을 하는데 만약 끝점에 도달했다면, 해당 출발점에서 얻을 수 있는 경로를 그것으로 fix하고 다음 줄로 넘어간다. 같은 점의 다른 경로는 현재 경로보다 무조건 이후에 도움이 되지 않기 때문이다.

 

- 해당 사고가 어찌보면 당연한 것이지만, 문제를 보자마자 완탐의 생각밖에 하지 못하였다.

 

- 또 어려웠던 부분은, 탐색한 경로를 원복하지 않아도 되는 것이었다. 윗줄부터 가장 유망한 경로들을 위주로 탐색했으므로, 밑줄에서는 위에서 탐색한 경로를 갈 일이 없기 때문이다. 따라서 탐색한 경로들에 대해서 따로 원복을 하지 않아도 된다.

 

- 대부분의 이런 문제들은 탐색을 한 후, 원복을 하고 다시 탐색을 하는 유형이 많은데, 이것이 매우 생소한 부분이었다.

 

 

 

자바 ver.

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

public class Main {
    static int r,c;
    static int[] dx={-1,0,1};
    static int[] dy={1,1,1};
    static int tmp;

    static char[][] arr;

    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        st=new StringTokenizer(br.readLine());

        r=Integer.parseInt(st.nextToken()); //세로
        c=Integer.parseInt(st.nextToken()); //가로

        arr=new char[r][c];

        for(int i=0; i<r; i++){
            String s = br.readLine();
            for(int j=0; j<c; j++){
                arr[i][j]=s.charAt(j);
            }
        }

        tmp=0;

        for(int i=0; i<r; i++){
            dfs(i,0);
        }

        System.out.println(tmp);

    }


    public static boolean dfs(int x, int y){

        arr[x][y]='o';

        if(y==c-1){
            tmp++;
            return true;
        }

        for(int i=0; i<3; i++){
            int nx=x+dx[i];
            int ny=y+dy[i];

            if(nx<0 || nx>=r || ny<0 || ny>=c){
                continue;
            }

            if(arr[nx][ny]=='.'){
                if(dfs(nx,ny)){
                    return true;
                }
            }
        }
        return false;
    }
}