inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

자바 코딩테스트 - it 대기업 유제

1. "최소 비행료" 문제 해법

우선순위 큐를 활용해도 되나요?

465

jhjikhsdsdw

작성한 질문수 27

0

우선순위 큐를 써서 문제를 풀었는데 채점 지원이 안되다 보니 제 논리에 허점이 있나 확인차 질문드립니다.

도착지 정보와 현재 지점까지 오는데 걸리는 비용, 환승 횟수 정보를 가지고 있는 Path 클래스를 이용했습니다.

static class Path implements Comparable<Path>{
    int end, cost, t;

    public Path(int end, int cost, int t) {
        this.end = end;
        this.cost = cost;
        this.t = t;
    }

    @Override
    public int compareTo(Path p) {
        return this.cost - p.cost;
    }
}

public int solution(int n, int[][] flights, int s, int e, int k){
    int answer = 0;

    ArrayList<ArrayList<Path>> list = new ArrayList<>();
    for(int i = 0; i < n; i++) {
        list.add(new ArrayList<>());
    }

    for (int[] flight : flights) {
        int start = flight[0];
        int end = flight[1];
        int cost = flight[2];
        list.get(start).add(new Path(end, cost, 0));
    }

    PriorityQueue<Path> pq = new PriorityQueue<>();
    for (Path path : list.get(s)) {
        pq.add(path);
    }

    while(!pq.isEmpty()) {
        Path cur = pq.poll();
        if(cur.t <= k && cur.end == e) {
            answer = cur.cost;
            break;
        }
        for (Path path : list.get(cur.end)) {
            pq.add(new Path(path.end, path.cost + cur.cost, cur.t + 1));
        }
    }

    return answer == 0 ? -1 : answer;
}

 클래스를 만들지 않고 List<int[]> 로 만드시는 이유도 궁금합니다. 배열로 만들면 int[0]이 무엇이 의미하는지 알기 힘들지만, 클래스로 만들면 end, cost 등 변수 이름으로 표현하기더 쉬워서 좋은 것 같은데 알고리즘에는 부적합한가요?

java 코딩-테스트

답변 1

1

김태원

안녕하세요^^

  1. 잘 하신 코드입니다. 다만 무한루프에 빠질 수 있는 코드입니다. 아래 입력의 경우 무한루프에 빠지게 되는 입력입니다.

System.out.println(T.solution(4, new int[][]{{0, 3, 12},{2, 0, 15}, {3, 1, 16}, {1, 3, 16}}, 3, 0, 3));

 

  1. 코테를 실제 시험보다 보면 시간이 부족할 때가 많습니다. 그래서 그냥 클래스를 안 만들고 바로 배열로 저는 합니다. 별다른 이유는 없습니다.

 

비밀번호

0

69

1

과일 가져가기 이러한 경우에는 반례가 생기지 않나요?

0

164

2

cpu 스케줄링

0

107

2

외부 문제 질문

0

122

2

가장 많이 사용된 회의실

0

118

2

심사위원 문제 시간복잡도 질문

0

127

1

현관문 출입순서

0

98

1

미로의 최단거리 통로

0

74

1

집으로 이동 문제 코드

0

125

1

채점 사이트 개설

0

161

2

송아지를 잡자

1

110

1

다익스트라 + 환승횟수

0

135

2

문제풀이 해설 질문입니다.

0

124

2

"이동 횟수" 문제가 변형된다면?

0

156

2

예제 3번의 정답이 이해가 되지 않아요 선생님 ㅜㅜ

0

249

1

"비밀번호" 문제 확인 부탁드립니다!

0

171

1

최대 길이 연속수열 질문

0

193

1

잃어버린 강아지 문제 count 관련 질문있습니다

0

204

1

바둑대회 질문입니당

0

222

1

5. "최대 길이 바이토닉 수열" 에서 설명해주신 방법과 제가 직접 구현한 방법이 달라, 확인 한번 부탁드립니다

0

311

1

알파코드 풀이질문입니다

0

218

1

7번 비밀 번호 문제에 시간복잡도가 궁금합니다!

0

164

1

혹시 이렇게 작성해도 괜찮나요?

0

287

2

문제풀이 확인 부탁드립니다.

0

246

1