https://www.acmicpc.net/problem/1753
import java.io.*;
import java.util.*;
class Node implements Comparable<Node> {
int idx;
int cost;
// 생성자
Node(int idx, int cost) {
this.idx = idx;
this.cost = cost;
}
@Override
public int compareTo(Node other) {
return Integer.compare(this.cost, other.cost);
}
}
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int V = Integer.parseInt(st.nextToken());
int E = Integer.parseInt(st.nextToken());
int start = Integer.parseInt(br.readLine());
// 인접 리스트 사용
List<List<Node>> graph = new ArrayList<>();
for (int i = 0; i <= V; i++) {
graph.add(new ArrayList<>()); //삽입
}
// N개의 경로 입력
for (int i = 0; i < E; i++) {
st = new StringTokenizer(br.readLine());
int x = Integer.parseInt(st.nextToken()); //시작
int y = Integer.parseInt(st.nextToken()); //도착
int length = Integer.parseInt(st.nextToken()); //길이(가중치)
graph.get(x).add(new Node(y, length)); // x에서 y로 가는 간선
}
// 최단 거리 저장 배열
int[] dist = new int[V+1];
Arrays.fill(dist, Integer.MAX_VALUE);
dist[start] = 0; // 출발지의 거리 초기화
// 우선순위 큐 사용
PriorityQueue<Node> q = new PriorityQueue<>();
q.add(new Node(start, 0)); //출발지 정점과 가중치를 우선순위 큐에 넣음
// // 방문 여부를 체크할 배열
// boolean[] visited = new boolean[N + 1];
while (!q.isEmpty()) { //큐에 값이 하나도 없을때까지 반복
Node now = q.poll(); //큐에서 앞에 있는 노드 반환 및 삭제 (현재 큐에 있는 값 중 출발지로부터 가장 가까운 거리)
int nowIdx = now.idx;
// if (visited[nowIdx]) //방문했던 노드라면 패스
// continue;
// visited[nowIdx] = true; //방문했다고 표시
if (dist[nowIdx] < now.cost) { //해당 노드에 대한 distance값은 여러번 갱신될 수 있어서 큐에 자주 들어갈 수있지만, 한번 방문됐다면 distance의 값은 최소값이다.
continue;//distance의 값이 더 작으니까 볼 필요없다 -> 이미 방문된 노드이다.
}
for (Node neighbor : graph.get(nowIdx)) {
if (dist[neighbor.idx] > (dist[nowIdx] + neighbor.cost)) {
dist[neighbor.idx] = dist[nowIdx] + neighbor.cost;
q.add(new Node(neighbor.idx, dist[neighbor.idx]));
}
}
}
for(int i=1; i<=V; i++) {
if (dist[i] == Integer.MAX_VALUE) {
System.out.println("INF");
} else {
System.out.println(dist[i]);
}
}
}
}