95 lines
4.4 KiB
Markdown
95 lines
4.4 KiB
Markdown
|
---
|
|||
|
title: Dijkstra's Algorithm
|
|||
|
localeTitle: Алгоритм Дейкстры
|
|||
|
---
|
|||
|
# Алгоритм Дейкстры
|
|||
|
|
|||
|
Алгоритм Дейкстры - это алгоритм графа, представленный Е. В. Дейкстром. Он находит кратчайший путь одного источника в графе с неотрицательными ребрами (почему?)
|
|||
|
|
|||
|
Мы создаем 2 массива: посещенные и удаленные, которые фиксируют, посещена ли вершина и каково минимальное расстояние от исходной вершины соответственно. Первоначально посещаемый массив присваивается как ложный и дальний как бесконечный.
|
|||
|
|
|||
|
Мы начинаем с исходной вершины. Пусть текущая вершина равна u, а ее смежные вершины - v. Теперь для каждого v, смежного с u, расстояние обновляется, если оно не было посещено раньше, а расстояние от u меньше его текущего расстояния. Затем мы выбираем следующую вершину с наименьшим расстоянием и которая не была посещена.
|
|||
|
|
|||
|
Приоритетная очередь часто используется для удовлетворения этого последнего требования за минимальное время. Ниже приведена реализация одной и той же идеи с использованием очереди приоритетов в Java.
|
|||
|
|
|||
|
```java
|
|||
|
import java.util.*;
|
|||
|
public class Dijkstra {
|
|||
|
class Graph {
|
|||
|
LinkedList<Pair<Integer>> adj[];
|
|||
|
int n; // Number of vertices.
|
|||
|
Graph(int n) {
|
|||
|
this.n = n;
|
|||
|
adj = new LinkedList[n];
|
|||
|
for(int i = 0;i<n;i++) adj[i] = new LinkedList<>();
|
|||
|
}
|
|||
|
// add a directed edge between vertices a and b with cost as weight
|
|||
|
public void addEdgeDirected(int a, int b, int cost) {
|
|||
|
adj[a].add(new Pair(b, cost));
|
|||
|
}
|
|||
|
public void addEdgeUndirected(int a, int b, int cost) {
|
|||
|
addEdgeDirected(a, b, cost);
|
|||
|
addEdgeDirected(b, a, cost);
|
|||
|
}
|
|||
|
}
|
|||
|
class Pair<E> {
|
|||
|
E first;
|
|||
|
E second;
|
|||
|
Pair(E f, E s) {
|
|||
|
first = f;
|
|||
|
second = s;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
// Comparator to sort Pairs in Priority Queue
|
|||
|
class PairComparator implements Comparator<Pair<Integer>> {
|
|||
|
public int compare(Pair<Integer> a, Pair<Integer> b) {
|
|||
|
return a.second - b.second;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
// Calculates shortest path to each vertex from source and returns the distance
|
|||
|
public int[] dijkstra(Graph g, int src) {
|
|||
|
int distance[] = new int[gn]; // shortest distance of each vertex from src
|
|||
|
boolean visited[] = new boolean[gn]; // vertex is visited or not
|
|||
|
Arrays.fill(distance, Integer.MAX_VALUE);
|
|||
|
Arrays.fill(visited, false);
|
|||
|
PriorityQueue<Pair<Integer>> pq = new PriorityQueue<>(100, new PairComparator());
|
|||
|
pq.add(new Pair<Integer>(src, 0));
|
|||
|
distance[src] = 0;
|
|||
|
while(!pq.isEmpty()) {
|
|||
|
Pair<Integer> x = pq.remove(); // Extract vertex with shortest distance from src
|
|||
|
int u = x.first;
|
|||
|
visited[u] = true;
|
|||
|
Iterator<Pair<Integer>> iter = g.adj[u].listIterator();
|
|||
|
// Iterate over neighbours of u and update their distances
|
|||
|
while(iter.hasNext()) {
|
|||
|
Pair<Integer> y = iter.next();
|
|||
|
int v = y.first;
|
|||
|
int weight = y.second;
|
|||
|
// Check if vertex v is not visited
|
|||
|
// If new path through u offers less cost then update distance array and add to pq
|
|||
|
if(!visited[v] && distance[u]+weight<distance[v]) {
|
|||
|
distance[v] = distance[u]+weight;
|
|||
|
pq.add(new Pair(v, distance[v]));
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
return distance;
|
|||
|
}
|
|||
|
|
|||
|
public static void main(String args[]) {
|
|||
|
Dijkstra d = new Dijkstra();
|
|||
|
Dijkstra.Graph g = d.new Graph(4);
|
|||
|
g.addEdgeUndirected(0, 1, 2);
|
|||
|
g.addEdgeUndirected(1, 2, 1);
|
|||
|
g.addEdgeUndirected(0, 3, 6);
|
|||
|
g.addEdgeUndirected(2, 3, 1);
|
|||
|
g.addEdgeUndirected(1, 3, 3);
|
|||
|
|
|||
|
int dist[] = d.dijkstra(g, 0);
|
|||
|
System.out.println(Arrays.toString(dist));
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
```
|