
    Gd                     N    d Z ddlmZ ddlmZmZ ddlmZ dgZ	ej
        Zd	dZdS )
z1Functions related to the Wiener index of a graph.    )chain   )is_connectedis_strongly_connected)shortest_path_lengthwiener_indexNc           	         |                                  }|rt          |           r|st          |           st          d          S t	          t          d t          | |          D                                 }|r|n|dz  S )a  Returns the Wiener index of the given graph.

    The *Wiener index* of a graph is the sum of the shortest-path
    distances between each pair of reachable nodes. For pairs of nodes
    in undirected graphs, only one orientation of the pair is counted.

    Parameters
    ----------
    G : NetworkX graph

    weight : object
        The edge attribute to use as distance when computing
        shortest-path distances. This is passed directly to the
        :func:`networkx.shortest_path_length` function.

    Returns
    -------
    float
        The Wiener index of the graph `G`.

    Raises
    ------
    NetworkXError
        If the graph `G` is not connected.

    Notes
    -----
    If a pair of nodes is not reachable, the distance is assumed to be
    infinity. This means that for graphs that are not
    strongly-connected, this function returns ``inf``.

    The Wiener index is not usually defined for directed graphs, however
    this function uses the natural generalization of the Wiener index to
    directed graphs.

    Examples
    --------
    The Wiener index of the (unweighted) complete graph on *n* nodes
    equals the number of pairs of the *n* nodes, since each pair of
    nodes is at distance one::

        >>> n = 10
        >>> G = nx.complete_graph(n)
        >>> nx.wiener_index(G) == n * (n - 1) / 2
        True

    Graphs that are not strongly-connected have infinite Wiener index::

        >>> G = nx.empty_graph(2)
        >>> nx.wiener_index(G)
        inf

    infc              3   D   K   | ]\  }}|                                 V  d S N)values).0vps      :lib/python3.11/site-packages/networkx/algorithms/wiener.py	<genexpr>zwiener_index.<locals>.<genexpr>J   s.      DDdaqxxzzDDDDDD    )weight   )is_directedr   r   floatsumchainispl)Gr   r   totals       r   r   r      s    l --//K 1!44  ,Q U||DDc!F.C.C.CDDDDDEEE.55UQY.r   r   )__doc__	itertoolsr   
componentsr   r   shortest_pathsr   r   __all__from_iterabler   r    r   r   <module>r$      sw    7 7       ; ; ; ; ; ; ; ; 7 7 7 7 7 7
 
	=/ =/ =/ =/ =/ =/r   