zoukankan      html  css  js  c++  java
  • 指派问题(匈牙利算法)

    问题描述:

    在生活中经常遇到这样的问题,某单位需完成n项任务,恰好有n个人可承担这些任务。由于每人的专长不同,各人完成任务不同(或所费时间),效率也不同。于是产生应指派哪个人去完成哪项任务,使完成n项任务的总效率最高(或所需总时间最小)。这类问题称为指派问题或分派问题。
    指派问题也是0-1规划,线性规划用到的是官网scipy.optimize库函数。
    示例: cost matrix = [ [1  4 3], [2 0 5], [3 2 2]]
    python 解决方案中,用到的是scipy.optimize.linear_sum_assignment(cost_matrix)
    代码实现:
    from scipy.optimize import linear_sum_assignment
     
    cost =np.array([[4,1,3],[2,0,5],[3,2,2]])
    row_ind,col_ind=linear_sum_assignment(cost)
    print(row_ind)#开销矩阵对应的行索引
    print(col_ind)#对应行索引的最优指派的列索引
    print(cost[row_ind,col_ind])#提取每个行索引的最优指派列索引所在的元素,形成数组
    print(cost[row_ind,col_ind].sum())#数组求和  

    输出:
    [0 1 2]

    [1 0 2] 

    [1 2 2] 

    5
  • 相关阅读:
    bzoj 1051: [HAOI2006]受欢迎的牛
    bzoj 1192: [HNOI2006]鬼谷子的钱袋
    一些动规水题
    USACO 2014 Open Silver Fairphoto
    USACO 2013 Nov Silver Pogo-Cow
    09day1
    09day2
    08day2
    08day1
    07day2
  • 原文地址:https://www.cnblogs.com/ylHe/p/9287384.html
Copyright © 2011-2022 走看看