python实现Huntington的5种除数法

HUntington的5种除数法分别是:
1、 最大除数法(GD)
2、 主要分数法(MF)
3、 相等比例法(EP)
4、 调和平均法(HM)
5、 最小除数法(SD)

from math import floor
from math import sqrt

data=[9061,7179,5259,3319,1182]   #A,B,C,D,E各地区人数
N=26         #总席位26

class struct:            # 地区人数情况
    def __init__(self, p, n):
        self.p = p    #地区总人数
        self.n = n    #座席数
        self.q = 0    #不公平度

def findmax(R):    #找不公平度最高的地区
    max = 0
    for i in range(1, len(R)):
        if R[i].q > R[max].q:
            max = i
    return max

#每种除法的除数d(n)
def GD_dn(n):   
    return n+1
def MF_dn(n):
    return n+0.5
def EP_dn(n):
    return sqrt(n*(n+1))
def HM_dn(n):
    return (2*n*(n+1))/(2*n+1)
def SD_dn(n):
    return n

def Huntington(data,N,dn):
    R = []                                     
    for i in range(0, len(data)):                   #给每个选区先分配一个席位
        R.append(struct(data[i], 1))
        R[i].q = R[i].p / dn(R[i].n)   #计算当前每个地区获得一个席位后的不公平度

    for i in range(0, N - len(data)):               #分配剩下的所有席位
        max = findmax(R)
        R[max].n = R[max].n + 1 
        R[max].q = R[max].p / dn(R[max].n)   #计算获得席位后的不公平度

    answer = []
    for i in range(0, len(data)):
        answer.append(R[i].n)
    print(answer)

def menu():
    print("                         菜单")
    print("                                                    ")
    print("                  1.最大除数法(GD)")
    print("                                                    ")
    print("                  2.主要分数法(MF)")
    print("                                                    ")
    print("                  3.相等比例法(EP)")
    print("                                                    ")
    print("                  4.调和平均法(HM)")
    print("                                                    ")
    print("                  5.最小除数法(SD)")
    print("                                                    ")
    print("                  6.退出(0退出)")
    flag = 1
    while flag != 0:
        flag = int(input("输入选项:"))
        if flag == 0:
            break
        elif flag == 1 :
            Huntington(data,N,GD_dn)
        elif flag == 2 :
            Huntington(data,N,MF_dn)
        elif flag == 3 :
            Huntington(data,N,EP_dn)
        elif flag == 4 :
            Huntington(data,N,HM_dn)
        elif flag == 5 :
            Huntington(data,N,SD_dn)
menu()



你可能感兴趣的:(python实现Huntington的5种除数法)