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()