백준 14620. 꽃길 :: 돼지개발자

출저 : https://www.acmicpc.net/problem/14620

"재귀를 활용한 완전 탐색"

가능한 씨앗의 배치를 모두 검색하는 문제이다. 씨앗을 중심으로 4방향이 화단에 들어오면서, 겹치지 않는 경우에 따라 세 개의 꽃을 배치하고 해당 위치의 가격을 모두 더한 값 중에 최소값이 답이다. 꽃잎 배치의 경우 가장자리에선 꽃을 피지 못하므로 1 ~  N-2 까지 탐색한다.

DFS에서 x 좌표를 주어 이전 씨앗부터 탐색한다. (y 좌표도 넣어주면 더 좋지만... N이 크지 않기에 생략)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
    static int N;
    static int[][] cost;
    static int min = Integer.MAX_VALUE;
    static boolean[][] visited;
    static int[] dx = {0,0,1,-1};
    static int[] dy = {1,-1,0,0};
    
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        N = Integer.parseInt(br.readLine());
        
        cost = new int[N][N];
        visited = new boolean[N][N];
        
        for (int i = 0; i < N; i++) {
            StringTokenizer  st = new StringTokenizer(br.readLine());
            for (int j = 0; j < N; j++) {
                cost[i][j] = Integer.parseInt(st.nextToken());
            }
        }
        
        solve(0,0,1);
        System.out.println(min);
    }
    
    static void solve(int cnt, int sum, int x) {
        if(cnt == 3) {
            min = Math.min(min, sum);
            return;
        }
        
        for (int i = x; i < N-1; i++) {
            for (int j = 1; j < N-1; j++) {
                if(visited[i][j]) 
                    continue;
                if(!available(i, j)) 
                    continue;
                int c = cost[i][j];
                visited[i][j] = true;
                
                for (int k = 0; k < 4; k++) {
                    visited[i+dx[k]][j+dy[k]] = true;
                    c += cost[i+dx[k]][j+dy[k]];
                }
                solve(cnt+1, sum +c, i);
                
                visited[i][j] = false;
                for (int k = 0; k < 4; k++) {
                    visited[i+dx[k]][j+dy[k]] = false;
                }
            }
        }
    }
    
    static boolean available(int x, int y) {
        for (int i = 0; i < 4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            
            if(x < 0 ||  x >= N || y < 0 || y >= N || visited[nx][ny])
                return false;
        }
        return true;
    }
}