問題
ある男女10人のグループがあり、男女間には好意があると仮定します。
このグループを、男女のペアを作ることができるようにマッチングさせることを考えます。
ただし、それぞれの男性・女性が同時に複数の異性とのペアを持てないものとします。
以下の男女間の好意度が与えられています。
この好意度を最大化するように、マッチングを決定してください。
男1 | 男2 | 男3 | 男4 | 男5 | |
---|---|---|---|---|---|
女1 | 70 | 90 | 80 | 80 | 70 |
女2 | 60 | 10 | 80 | 70 | 60 |
女3 | 70 | 70 | 100 | 70 | 60 |
女4 | 80 | 80 | 80 | 50 | 60 |
女5 | 90 | 90 | 90 | 90 | 30 |
解法
NetworkXライブラリを使ってグラフを作成します。
このグラフのノードは男女それぞれの番号で表されます。
辺には好意度に対応する重みが付いています。
1 | import networkx as nx |
NetworkXライブラリを用いて、与えられたグラフの最大重みマッチングを求めています。
まず、nx.Graph()を使って空のグラフを定義し、Gという変数に格納しています。
次に、add_nodes_fromを使って、グラフのノードを追加しています。
ここでは、[“F1”, “F2”, “F3”, “F4”, “F5”]と[“M1”, “M2”, “M3”, “M4”, “M5”]という2つの集合を作り、それぞれに対してbipartite=0とbipartite=1という属性を付与しています。
これによって、グラフが二部グラフであることを示しています。
その後、add_weighted_edges_fromを使って、グラフの辺を追加しています。
ここでは、各辺に対して、始点、終点、重みを設定しています。
最後に、nx.max_weight_matchingを使って、最大重みマッチングを求め、結果をmax_matchingに格納しています。
[実行結果]
{('F4', 'M1'), ('F1', 'M2'), ('M4', 'F5'), ('M3', 'F3'), ('F2', 'M5')}
最大重みマッチングによって、以上のような男女のペアを作ることができました。