1. 문제 설명
- 설명
N×N 크기의 도시지도가 있습니다. 도시지도는 1×1크기의 격자칸으로 이루어져 있습니다.
각 격자칸에는 0은 빈칸, 1은 집, 2는 피자집으로 표현됩니다. 각 격자칸은 좌표(행번호, 열 번호)로 표현됩니다.
행번호는 1번부터 N번까지이고, 열 번호도 1부터 N까지입니다.
도시에는 각 집마다 “피자배달거리”가 았는데 각 집의 피자배달거리는 해당 집과 도시의 존재하는
피자집들과의 거리 중 최소값을 해당 집의 “피자배달거리”라고 한다.
집과 피자집의 피자배달거리는 |x1-x2|+|y1-y2| 이다.
예를 들어, 도시의 지도가 아래와 같다면
(1, 2)에 있는 집과 (2, 3)에 있는 피자집과의 피자 배달 거리는 |1-2| + |2-3| = 2가 된다.
최근 도시가 불경기에 접어들어 우후죽순 생겼던 피자집들이 파산하고 있습니다.
도시 시장은 도시에 있는 피자집 중 M개만 살리고 나머지는 보조금을 주고 폐업시키려고 합니다.
시장은 살리고자 하는 피자집 M개를 선택하는 기준으로 도시의 피자배달거리가 최소가 되는 M개의 피자집을 선택하려고 합니다.
도시의 피자 배달 거리는 각 집들의 피자 배달 거리를 합한 것을 말합니다. - 입력
첫째 줄에 N(2 ≤ N ≤ 50)과 M(1 ≤ M ≤ 12)이 주어진다.
둘째 줄부터 도시 정보가 입력된다. - 출력
첫째 줄에 M개의 피자집이 선택되었을 때 도시의 최소 피자배달거리를 출력한다. - 출처
삼성 SW역량평가 기출문제 : DFS 활용
2. 문제 코드 및 풀이 설명
import java.util.ArrayList;
import java.util.Scanner;
class Point {
public int x, y;
Point(int x, int y) {
this.x = x;
this.y = y;
}
}
public class PizzaDelivery {
static int n, m, len, answer = Integer.MAX_VALUE;
static int[] combi;
static ArrayList<Point> hs, pz;
public static void main(String[] args) {
PizzaDelivery T = new PizzaDelivery();
Scanner kb = new Scanner(System.in);
n = kb.nextInt();
m = kb.nextInt();
pz = new ArrayList<>(); // 피자 가게의 위치
hs = new ArrayList<>(); // 집의 위치
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
int tmp = kb.nextInt();
if (tmp == 1) hs.add(new Point(i, j)); // 집을 입력
else if (tmp == 2) pz.add(new Point(i, j)); // 피자 가게를 입력
}
}
len = pz.size(); // 입력된 피자 집의 수
combi = new int[m]; // 피자 집의 조합
T.DFS(0, 0);
System.out.println(answer);
}
// 입력된 피자 집의 조합을 구함
// 입력된 피자 집을 남겨야하는 피자 집의 수만큼으로 조합을 하여 거리를 구한다
// 구한 거리가 기존의 answer보다 짧다면 바꾸어준다.
public void DFS(int L, int s) {
if (L == m) {
int sum = 0;
for(Point h : hs){
int dis = Integer.MAX_VALUE;
for (int i : combi){ // 입력된 조합에서의 피자배달거리를 구한다
dis = Math.min(dis, Math.abs(h.x-pz.get(i).x) + Math.abs(h.y - pz.get(i).y));
}
sum += dis; // 구한 거리를 입력
}
answer = Math.min(answer, sum); // 구한 값이 기존 값보다 작다면 변경
} else {
// 모든 조합을 확인하도록 반복문
for (int i = s; i < len; i++) {
combi[L] = i; // 현재 조합에 입력
DFS(L + 1, i + 1); // 입력될 다음 수를 찾으러
}
}
}
}
3. 출력 예시
입력
4 4
0 1 2 0
1 0 2 1
0 2 1 2
2 0 1 2
출력
6
위의 내용은 인프런에서 수강할 수 있는 김태원님의 자바 알고리즘 문제풀이 강의를 바탕으로 공부한 내용을 정리한 내용입니다!
자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의
자바(Java)로 코딩테스트를 준비하시는 분을 위한 강좌입니다. 코딩테스트에서 가장 많이 출제되는 Top 10 Topic을 다루고 있습니다. 주제와 연동하여 기초문제부터 중급문제까지 단계적으로 구성
www.inflearn.com