;i
@ d Z ddlmZ ddZd Z G d de ZdS ) zM
altgraph.GraphAlgo - Graph algorithms
=====================================
)
GraphErrorNc t i }i }t }d||< |D ]}|| ||< ||k r n| | D ]p}| || }|| | | z } ||v r | || k rt d V||vs | || k r
| ||< |||< q||fS )a`
Dijkstra's algorithm for shortest paths
`David Eppstein, UC Irvine, 4 April 2002
`_
`Python Cookbook Recipe
`_
Find shortest paths from the start node to all nodes nearer than or
equal to the end node.
Dijkstra's algorithm is only guaranteed to work correctly when all edge
lengths are positive. This code does not verify this property for all
edges (only the edges examined until the end vertex is reached), but will
correctly compute shortest paths even for some graphs with negative edges,
and will raise an exception if it discovers that a negative edge has
caused it to make a mistake.
Adapted to altgraph by Istvan Albert, Pennsylvania State University -
June, 9 2004
r z3Dijkstra: found better path to already-final vertex)_priorityDictionaryout_nbrs edge_by_node edge_datar )
graphstartendDPQvwedge_idvwLengths
FC:\PYTHON\MyICR_Workspace\venv\Lib\site-packages\altgraph/GraphAlgo.pydijkstrar s . A
A AAeH
t! 88 E ""
A((A..Gteoog 6 66HAvvad??$M # !x!A$ !!
q6M c t | || \ }}g } | | ||k rn || }%| |S )a
Find a single shortest path from the *start* node to the *end* node.
The input has the same conventions as dijkstra(). The output is a list of
the nodes in order along the shortest path.
**Note that the distances must be stored in the edge data as numeric data**
)r appendreverse)r r
r r r
Paths r
shortest_pathr 8 sc E5# & &DAq
D C %<< f
LLNNN Kr c 0 e Zd ZdZd Zd Zd Zd Zd ZdS )r a
Priority dictionary using binary heaps (internal use only)
David Eppstein, UC Irvine, 8 Mar 2002
Implements a data structure that acts almost like a dictionary, with
two modifications:
1. D.smallest() returns the value x minimizing D[x]. For this to
work correctly, all values D[x] stored in the dictionary must be
comparable.
2. iterating "for x in D" finds and removes the items from D in sorted
order. Each item is not removed until the next item is requested,
so D[x] will still return a useful value until the next iteration
of the for-loop. Each operation takes logarithmic amortized time.
c H g | _ t | dS )a
Initialize priorityDictionary by creating binary heap of pairs
(value,key). Note that changing or removing a dict entry will not
remove the old pair from the heap until it is found by smallest()
or until the heap is rebuilt.
N)_priorityDictionary__heapdict__init__)selfs r r z _priorityDictionary.__init__b s#
dr c D t | dk rt d | j }|d d | vs$| |d d |d d k r| }d} d|z dz }|dz t | k r|| ||dz k r|dz
}|t | k s ||| k r|||< n|| ||< |}k|d d | v| |d d |d d k |d d S )zU
Find smallest item after removing deleted items from front of heap.
r z$smallest of empty priorityDictionary )len
IndexErrorr pop)r heaplastIteminsertionPoint
smallChilds r smallestz _priorityDictionary.smallestl sB t99>> CDD D{1gaj$$T!WQZ(8DGAJ(F(FxxzzH N
, /! 3
NSYY.. Z(4
Q+??? !OJ T**h$z:J.J.J+3D ('+J'7^$!+
, 1gaj$$T!WQZ(8DGAJ(F(F Awqzr c " fd} | S )zK
Create destructive sorted iterator of priorityDictionary.
c 3 K t dk r0 } | V | = t dk .d S d S )Nr )r$ r+ )xr s r iterfnz,_priorityDictionary.__iter__..iterfn sL d))a--MMOOG d))a------r )r r/ s ` r __iter__z _priorityDictionary.__iter__ s)
vxxr c t | || | j }t | dt | z k r>d | D | _ | j dS ||f}t | }| d |dk rC|||dz
dz k r1||dz
dz ||< |dz
dz }|dk r|||dz
dz k 1|||< dS )z
Change value stored in dictionary and add corresponding pair to heap.
Rebuilds the heap if the number of deleted items gets large, to avoid
memory leakage.
r# c g | ] \ }}||f
S r0 r0 ).0kr s r
z3_priorityDictionary.__setitem__.. s ;;;daAq6;;;r Nr r" )r __setitem__r r$ itemssortr )r keyvalr' newPairr) s r r7 z _priorityDictionary.__setitem__ s!
sC((({ t99q3t99} $ $;;djjll;;;DK K CjG YYN KK 1$$4 !9KPQ8Q3R)R)R'+^a-?A,E'F^$"01"4!: !1$$4 !9KPQ8Q3R)R)R $+D r c $ || vr|| |< | | S )zT
Reimplement setdefault to pass through our customized __setitem__.
r0 )r r: r; s r
setdefaultz _priorityDictionary.setdefault s d??DICyr N) __name__
__module__ __qualname____doc__r r+ r1 r7 r> r0 r r r r O si $ 0 + + +( r r )N)rB altgraphr r r r r r0 r r rD s - - - -` .\ \ \ \ \$ \ \ \ \ \r