zarrabeitia / munkres_sparse Goto Github PK

from munkres_sparse import munkres

def main():
    values = [
        (1, 1, 3),
        (1, 3, 4),
        (2, 1, 2),
        (2, 2, 1),
        (2, 3, 3),
        (3, 2, 4),
        (3, 3, 5)
    ]

    match_res = munkres(values)

    values_dict = dict(((i, j), v) for i, j, v in values)
    weight_res = sum(values_dict[p] for p in match_res)
    print len(match_res)
    print match_res
    print weight_res
    print('complete')
    pass

if __name__=='__main__':
    main()

3
[(1, 1), (2, 2), (3, 3)]
9
complete

3
[(1, 1), (2, 3), (3, 2)]
10