[BOJ] 백준 15686 - 치킨배달 (JAVA)

조합한 경우 process(); 함수에서 최소값을 찾는다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;
 
public class Main15686_치킨배달 {
    private static int N;
    private static int M;
    private static int[][] map;
    private static Pair[] set;
    private static ArrayList<Pair> list;
    private static ArrayList<Pair> pos;
    static int minNum = Integer.MAX_VALUE;
 
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        N = Integer.parseInt(st.nextToken()); // 배열의 크기
        M = Integer.parseInt(st.nextToken()); // 폐업시키지 않을 치킨집
        map = new int[N][N];
        list = new ArrayList<Pair>(); // 치킨집의 x,y 좌표 저장
        pos = new ArrayList<Pair>(); // 도시 x, y 정보 저장 -> process() 호출 후 사용
        set = new Pair[M]; // 치킨집에서 조합에 쓰일 배열
 
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            for (int j = 0; j < N; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
                if (map[i][j] == 2) {
                    list.add(new Pair(i, j));
                }
 
                if (map[i][j] == 1) {
                    pos.add(new Pair(i, j));
                }
            }
        }
        comb(0, 0);
        System.out.println(minNum);
    }
 
    public static void comb(int idx, int k) {
        if (idx == M) { 
            process();
            return;
        } else {
            for (int i = k; i < list.size(); i++) {
                set[idx] = new Pair(list.get(i).x, list.get(i).y);
                comb(idx + 1, i + 1);
            }
        }
    }
 
    public static void process() {
        int result = 0;
        for (int i = 0; i < pos.size(); i++) {
            int x = pos.get(i).x;
            int y = pos.get(i).y;
            int min = Integer.MAX_VALUE;
            for (int j = 0; j < set.length; j++) {
                int num = Math.abs(x - set[j].x) + Math.abs(y - set[j].y);
                min = min > num ? num : min;
            }
            result += min;
        }
        
        minNum = minNum > result ? result : minNum;
    }
 
    static class Pair {
        int x;
        int y;
 
        public Pair(int x, int y) {
            super();
            this.x = x;
            this.y = y;
        }
 
    }
}