U
    ?h/  ã                   @   sB   d Z ddlmZ ddlmZ ddgZedƒdd	d„ƒZd
d„ ZdS )a  Functions for finding node and edge dominating sets.

A `dominating set`_ for an undirected graph *G* with vertex set *V*
and edge set *E* is a subset *D* of *V* such that every vertex not in
*D* is adjacent to at least one member of *D*. An `edge dominating set`_
is a subset *F* of *E* such that every edge not in *F* is
incident to an endpoint of at least one edge in *F*.

.. _dominating set: https://en.wikipedia.org/wiki/Dominating_set
.. _edge dominating set: https://en.wikipedia.org/wiki/Edge_dominating_set

é   )Únot_implemented_foré   )Úmaximal_matchingÚmin_weighted_dominating_setÚmin_edge_dominating_setZdirectedNc                    sx   t ˆ ƒdkrtƒ S tƒ ‰‡ ‡‡fdd„}tˆ ƒ}‡ fdd„ˆ D ƒ}|rtt| ¡ |d\}}ˆ |¡ ||= ||8 }qBˆS )a[  Returns a dominating set that approximates the minimum weight node
    dominating set.

    Parameters
    ----------
    G : NetworkX graph
        Undirected graph.

    weight : string
        The node attribute storing the weight of an node. If provided,
        the node attribute with this key must be a number for each
        node. If not provided, each node is assumed to have weight one.

    Returns
    -------
    min_weight_dominating_set : set
        A set of nodes, the sum of whose weights is no more than `(\log
        w(V)) w(V^*)`, where `w(V)` denotes the sum of the weights of
        each node in the graph and `w(V^*)` denotes the sum of the
        weights of each node in the minimum weight dominating set.

    Notes
    -----
    This algorithm computes an approximate minimum weighted dominating
    set for the graph `G`. The returned solution has weight `(\log
    w(V)) w(V^*)`, where `w(V)` denotes the sum of the weights of each
    node in the graph and `w(V^*)` denotes the sum of the weights of
    each node in the minimum weight dominating set for the graph.

    This implementation of the algorithm runs in $O(m)$ time, where $m$
    is the number of edges in the graph.

    References
    ----------
    .. [1] Vazirani, Vijay V.
           *Approximation Algorithms*.
           Springer Science & Business Media, 2001.

    é    c                    s&   | \}}ˆ j |  ˆd¡t|ˆ ƒ S )z¼Returns the cost-effectiveness of greedily choosing the given
        node.

        `node_and_neighborhood` is a two-tuple comprising a node and its
        closed neighborhood.

        é   )ZnodesÚgetÚlen)Znode_and_neighborhoodÚvZneighborhood©ÚGZdom_setÚweight© úb/var/www/html/venv/lib/python3.8/site-packages/networkx/algorithms/approximation/dominating_set.pyÚ_costE   s    z*min_weighted_dominating_set.<locals>._costc                    s    i | ]}||ht ˆ | ƒB “qS r   )Úset)Ú.0r   ©r   r   r   Ú
<dictcomp>U   s      z/min_weighted_dominating_set.<locals>.<dictcomp>)Úkey)r
   r   ÚminÚitemsÚadd)r   r   r   ZverticesZneighborhoodsZdom_nodeZmin_setr   r   r   r      s    *

c                 C   s   | st dƒ‚t| ƒS )aã  Returns minimum cardinality edge dominating set.

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

    Returns
    -------
    min_edge_dominating_set : set
      Returns a set of dominating edges whose size is no more than 2 * OPT.

    Notes
    -----
    The algorithm computes an approximate solution to the edge dominating set
    problem. The result is no more than 2 * OPT in terms of size of the set.
    Runtime of the algorithm is $O(|E|)$.
    z"Expected non-empty NetworkX graph!)Ú
ValueErrorr   r   r   r   r   r   f   s    )N)Ú__doc__Úutilsr   Zmatchingr   Ú__all__r   r   r   r   r   r   Ú<module>   s   P