问题描述
给一个无向图,且图中不能包含权值为负数的路径。请求出任意一个节点到其他所有节点的最短路径长度。
思路分析
从任意一个节点出发,然后,此时,这个节点只有到自己的路径是0,到其他节点的路径还没解锁,都是无穷。这个无穷怎么表示?到某个节点的最短路径怎么表示?可以用map,key是节点,value是此时到某个节点的最短路径长度,无穷,可以使用直接不把某个节点放入map来表示,也就是说,某个节点不在map中就表示他是无穷。
然后此题就是在不断维护这个map,然后具体思路类似prim算法,每次从解锁的路径中找一个最短路径长度,然后更新这个路径到达的节点,把到达这个节点的最短路径更新,然后,到这个节点的最短路径就找到了。需要注意,每次需要看看这个路径连接的节点的最短路径是否已经找过了?如果已经找过的,没必要重复找,因此,我们还需要维护一个set,来保存到哪些节点的最短路径已经找到了,已经找到的,就依次找下一个最短路径。
因此,此题需要的数据结构:
HashMap<Node,Integer> distanceMap;
用来表示到某个节点的最短路径,每次会有一个节点的最终节点定下来,其余的会被不断替换更新,如果key不在里面,表示到当前节点的路径长度为无穷。
HashSet selected;
用来表示某个节点是否已经找到最短路径,已经找到的不能重复找。
代码
public static HashMap<Node,Integer> dijkstral(Node head){
HashMap<Node,Integer> distanceMap=new HashMap<>();
HashSet<Node> selected=new HashSet<>();
distanceMap.put(head,0);
Node node=getMinAndNotSeleted(distanceMap,selected);
while (node!=null){
int distance=distanceMap.get(node);
for (Edges edges: node.edges){
if(!distanceMap.containsKey(edges.to)){
distanceMap.put(edges.to,distance+edges.weight);
}else {
distanceMap.put(edges.to,Math.min(distanceMap.get(edges.to),distance+edges.weight));
}
}
selected.add(node);
node=getMinAndNotSeleted(distanceMap,selected);
}
return distanceMap;
}
//找没找过的节点的最短路径
public static Node getMinAndNotSeleted(HashMap<Node,Integer> distanceMap,HashSet<Node> selected){
int min=Integer.MAX_VALUE;
Node n = null;
for (Node node : distanceMap.keySet()){
if(distanceMap.get(node)<min&&!selected.contains(node)){
min=distanceMap.get(node);
n=node;
}
}
return n;
}