dijkstra

1 files changed, 20 insertions, 0 deletions

diff --git a/graph/dijkstra.cpp b/graph/dijkstra.cpp
new file mode 100644
index 0000000..e7ec0c3
--- /dev/null
+++ b/graph/dijkstra.cpp
@@ -0,0 +1,20 @@
+priority_queue<ii, vector<ii>, greater<ii> > pq;
+vector<int> dist;
+dist.assign(NUM_VERTICES, INF);
+dist[0] = 0;
+pq.push(ii(0, 0));
+
+while (!pq.empty()) {
+	di front = pq.top(); pq.pop();
+	int curNode = front.second, curDist = front.first;
+	
+	if (curDist > dist[curNode]) continue;
+	
+	for (i = 0; i < (int)adjlist[curNode].size(); i++) {
+		int nextNode = adjlist[curNode][i].first, nextDist = curDist + adjlist[curNode][i].second;
+		
+		if (nextDist < dist[nextNode]) {
+			dist[nextNode] = nextDist; pq.push(ii(nextDist, nextNode));
+		}
+	}
+}
\ No newline at end of file