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