Created
January 8, 2012 06:33
-
-
Save methodin/1577481 to your computer and use it in GitHub Desktop.
Prim's Algorithm
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// Represents an edge from source to sink with capacity | |
var Edge = function(source, sink, capacity) { | |
this.source = source; | |
this.sink = sink; | |
this.capacity = capacity; | |
}; | |
// Main class to manage the network | |
var Graph = function() { | |
this.edges = {}; | |
this.nodes = []; | |
this.nodeMap = {}; | |
// Add a node to the graph | |
this.addNode = function(node) { | |
this.nodes.push(node); | |
this.nodeMap[node] = this.nodes.length-1; | |
this.edges[node] = []; | |
}; | |
// Add an edge from source to sink with capacity | |
this.addEdge = function(source, sink, capacity) { | |
// Create the two edges = one being the reverse of the other | |
this.edges[source].push(new Edge(source, sink, capacity)); | |
this.edges[sink].push(new Edge(sink, source, capacity)); | |
}; | |
// Does edge from source to sink exist? | |
this.edgeExists = function(source, sink) { | |
if(this.edges[source] !== undefined) | |
for(var i=0;i<this.edges[source].length;i++) | |
if(this.edges[source][i].sink == sink) | |
return this.edges[source][i]; | |
return null; | |
}; | |
}; | |
function Prim(graph) { | |
var result = []; | |
var usedNodes = {}; | |
function findMin(g) { | |
var min = [999999,null]; | |
for(var i=0;i<result.length;i++) | |
for(var n=0;n<g.edges[result[i]].length;n++) | |
if(g.edges[result[i]][n].capacity < min[0] && usedNodes[g.edges[result[i]][n].sink] === undefined) | |
min = [g.edges[result[i]][n].capacity, g.edges[result[i]][n].sink]; | |
return min[1]; | |
} | |
// Pick random start point | |
var node = g.nodes[Math.round(Math.random()*(g.nodes.length-1))]; | |
result.push(node); | |
usedNodes[node] = true; | |
var min = findMin(g); | |
while(min != null) { | |
result.push(min); | |
usedNodes[min] = true; | |
min = findMin(g); | |
} | |
return result; | |
}; | |
var g = new Graph(); | |
g.addNode('a'); | |
g.addNode('b'); | |
g.addNode('c'); | |
g.addNode('d'); | |
g.addNode('e'); | |
g.addNode('f'); | |
g.addEdge('a','b',1); | |
g.addEdge('b','c',3); | |
g.addEdge('a','d',3); | |
g.addEdge('b','d',2); | |
g.addEdge('d','e',3); | |
g.addEdge('b','e',6); | |
g.addEdge('b','f',5); | |
g.addEdge('c','e',4); | |
g.addEdge('e','f',2); | |
g.addEdge('c','f',4); | |
var result = Prim(g); | |
document.write('<h2>Result</h2>'); | |
document.write(result); |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
What is the returned result? Is it an array of nodes or an array of edges?