[ BOJ / C++ ] 1753번 최단경로

황승환·2021년 7월 20일
0

C++

목록 보기
17/65

이번 문제는 다익스트라 알고리즘을 사용하여 해결하는 문제였다.
Dijkstra (다익스트라)

Code

보통 다익스트라 알고리즘은 배열로 구현하지만 이 문제의 메모리 제한 때문에 우선순위 큐로 구현하였다.

#include <iostream>
#include <vector>
#include <queue>
#define MAX 20010
#define INF 987654321
using namespace std;

int v,e,start;
int dist[MAX];
vector<pair<int, int>> map[MAX];

void Input(){
    cin>>v>>e>>start;
    for(int i=0; i<e; i++){
        int a,b,c;
        cin>>a>>b>>c;
        map[a].push_back(make_pair(b,c));
    }
    for(int i=1; i<=v; i++){
        dist[i]=INF;
    }
}

void Dijkstra(){
    priority_queue<pair<int, int>> pq;
    pq.push(make_pair(0, start));
    dist[start]=0;
    while(!pq.empty()){
        int cost=-pq.top().first;
        int cur=pq.top().second;
        pq.pop();
        for(int i=0; i<map[cur].size(); i++){
            int next=map[cur][i].first;
            int ncost=map[cur][i].second;
            
            if(dist[next]>cost+ncost){
                dist[next]=cost+ncost;
                pq.push(make_pair(-dist[next], next));
            }
        }
    }
    for(int i=1; i<=v; i++){
        if(dist[i]==INF)
            cout<<"INF"<<'\n';
        else
            cout<<dist[i]<<'\n';
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    Input();
    Dijkstra();
    return 0;
}

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글