우선순위 큐를 활용해도 되나요?
465
작성한 질문수 27
우선순위 큐를 써서 문제를 풀었는데 채점 지원이 안되다 보니 제 논리에 허점이 있나 확인차 질문드립니다.
도착지 정보와 현재 지점까지 오는데 걸리는 비용, 환승 횟수 정보를 가지고 있는 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 등 변수 이름으로 표현하기더 쉬워서 좋은 것 같은데 알고리즘에는 부적합한가요?
답변 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));
코테를 실제 시험보다 보면 시간이 부족할 때가 많습니다. 그래서 그냥 클래스를 안 만들고 바로 배열로 저는 합니다. 별다른 이유는 없습니다.
비밀번호
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





