mirror of
https://github.moeyy.xyz/https://github.com/trekhleb/javascript-algorithms.git
synced 2024-12-26 23:21:18 +08:00
Add Dijkstra.
This commit is contained in:
parent
8b057b10d0
commit
ce7a4a930f
@ -33,10 +33,17 @@ export default function dijkstra(graph, startVertex) {
|
|||||||
|
|
||||||
if (distanceToNeighborFromCurrent < existingDistanceToNeighbor) {
|
if (distanceToNeighborFromCurrent < existingDistanceToNeighbor) {
|
||||||
distances[neighbor.getKey()] = distanceToNeighborFromCurrent;
|
distances[neighbor.getKey()] = distanceToNeighborFromCurrent;
|
||||||
|
|
||||||
|
// Change priority.
|
||||||
|
if (queue.hasValue(neighbor)) {
|
||||||
|
queue.changePriority(neighbor, distances[neighbor.getKey()]);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Add neighbor to the queue for further visiting.
|
// Add neighbor to the queue for further visiting.
|
||||||
queue.add(neighbor, distances[neighbor.getKey()]);
|
if (!queue.hasValue(neighbor)) {
|
||||||
|
queue.add(neighbor, distances[neighbor.getKey()]);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
@ -40,19 +40,28 @@ export default class PriorityQueue extends MinHeap {
|
|||||||
* @return {PriorityQueue}
|
* @return {PriorityQueue}
|
||||||
*/
|
*/
|
||||||
changePriority(item, priority) {
|
changePriority(item, priority) {
|
||||||
const customFindingComparator = new Comparator((a, b) => {
|
this.remove(item, new Comparator(this.compareValue));
|
||||||
if (a === b) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
return a < b ? -1 : 1;
|
|
||||||
});
|
|
||||||
|
|
||||||
this.remove(item, customFindingComparator);
|
|
||||||
this.add(item, priority);
|
this.add(item, priority);
|
||||||
|
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param {*} item
|
||||||
|
* @return {Number[]}
|
||||||
|
*/
|
||||||
|
findByValue(item) {
|
||||||
|
return this.find(item, new Comparator(this.compareValue));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param {*} item
|
||||||
|
* @return {boolean}
|
||||||
|
*/
|
||||||
|
hasValue(item) {
|
||||||
|
return this.findByValue(item).length > 0;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @param {*} a
|
* @param {*} a
|
||||||
* @param {*} b
|
* @param {*} b
|
||||||
@ -65,4 +74,17 @@ export default class PriorityQueue extends MinHeap {
|
|||||||
|
|
||||||
return this.priorities[a] < this.priorities[b] ? -1 : 1;
|
return this.priorities[a] < this.priorities[b] ? -1 : 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @param {*} a
|
||||||
|
* @param {*} b
|
||||||
|
* @return {number}
|
||||||
|
*/
|
||||||
|
compareValue(a, b) {
|
||||||
|
if (a === b) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return a < b ? -1 : 1;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -87,4 +87,17 @@ describe('PriorityQueue', () => {
|
|||||||
expect(priorityQueue.poll()).toBe(15);
|
expect(priorityQueue.poll()).toBe(15);
|
||||||
expect(priorityQueue.poll()).toBe(10);
|
expect(priorityQueue.poll()).toBe(10);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('should be possible to search in priority queue by value', () => {
|
||||||
|
const priorityQueue = new PriorityQueue();
|
||||||
|
|
||||||
|
priorityQueue.add(10, 1);
|
||||||
|
priorityQueue.add(5, 2);
|
||||||
|
priorityQueue.add(100, 0);
|
||||||
|
priorityQueue.add(200, 0);
|
||||||
|
priorityQueue.add(15, 15);
|
||||||
|
|
||||||
|
expect(priorityQueue.hasValue(70)).toBeFalsy();
|
||||||
|
expect(priorityQueue.hasValue(15)).toBeTruthy();
|
||||||
|
});
|
||||||
});
|
});
|
||||||
|
Loading…
Reference in New Issue
Block a user