인프런 커뮤니티 질문&답변

vanillaa님의 프로필 이미지
vanillaa

작성한 질문수

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비

15. 피자배달거리(DFS)

시간 초과

작성

·

158

0

코드는 강의에 올려주신 것과 비슷한데 시간 초과가 발생합니다...

import java.util.*;

public class Ch8_14 {
    public static int n,m, dis, answer = Integer.MAX_VALUE;
    public static int[][] board;
    public static ArrayList<Point> pizza = new ArrayList<>();
    public static ArrayList<Point> house = new ArrayList<>();
    public static int[] combi;

    public static class Point {
        int x, y;
        Point(int x, int y) {
            this.x = x;
            this.y = y;
        }
    }

    public static void DFS(int index, int start) {
        if(index == m) {
            int sum = 0;
            for(Point h : house) {
                dis=Integer.MAX_VALUE;
                for (int idx : combi) {
                    dis = Math.min(dis, Math.abs(h.x - pizza.get(idx).x) + Math.abs(h.y - pizza.get(idx).y));
                }
                sum+=dis;
            }
            answer = Math.min(answer, sum);
        } else {
            for(int i = start; i < pizza.size(); i++) {
                combi[index] = i;
                DFS(index+1, start+1);
            }
        }
    }

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

        n = sc.nextInt();
        m = sc.nextInt();

        board = new int[n][n];

        // 입력 받기
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < n; j++) {
                int tmp = sc.nextInt();
                if(tmp == 1) house.add(new Point(i,j));
                if(tmp == 2) pizza.add(new Point(i,j));
            }
        }

        combi = new int[m];

        DFS(0,0);

        System.out.println(answer);
    }
}

답변 1

1

김태원님의 프로필 이미지
김태원
지식공유자

안녕하세요^^

재귀호출을 아래와 같이 해야 합니다.

DFS(index+1, i+1);

vanillaa님의 프로필 이미지
vanillaa

작성한 질문수

질문하기