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