Python實現環形連結串列

class Player:
    """節點類"""
    def __init__(self):
        """初始化姓名，分數，指標"""
        self.name = ''
        self.score = 0
        self.next = None
 
 
def ergodic(head, num=None, is_print=False):
    """遍歷函數，num是遍歷到哪一個位置序號，is_print是否觸發列印方法"""
    if head.next is None:
        return None
    ptr = head
    count = 0
    while True:
        count += 1
        if is_print:
            print('No.'+str(count), ptr.name, ptr.score, '--->', ptr.next.name)
        if count == num:
            break
        if ptr.next == head:
            break
        ptr = ptr.next
    return ptr  # 返回遍歷完成後的最後一個節點
 
 
def invert(x):  # x是連結串列的第一個節點
    """反轉環形連結串列"""
    y = x.next  # y是x原來的next
    x.next = ergodic(x)  # 將第一個節點的next指向最後一個節點(因為反轉了)
    while True:  # 迴圈反轉後面的所有節點
        r = y.next
        y.next = x
        if r == head:  # r是head說明y已經是原本連結串列的最後一個節點了
            return y  # 返回y，這個y是反轉後的連結串列的第一個節點
        x = y
        y = r
 
 
head = Player()
ptr = head
 
 
while True:
    select = input("(1).新增   (2).檢視   (3).插入   (4).刪除   (5).反轉   (6).離開n輸入：")
    if select == "1":  # 新增節點
        ptr = ergodic(head)  # 獲取當前連結串列最後一個節點
        if ptr is None:  # ptr為None說明當前在新增第一個節點head
            head.name = input("姓名：")
            head.score = input("分數：")
            head.next = head
        else:  # 新增第一個節點之後的節點
            next_data = Player()
            next_data.name = input("姓名：")
            next_data.score = input("分數：")
            next_data.next = head
            ptr.next = next_data
 
    elif select == "2":  # 遍歷檢視連結串列所有節點
        ergodic(head, is_print=True)  # 遍歷連結串列，將列印引數設為True
 
    elif select == '3':  # 向連結串列中任意位置插入節點，位置以序號表示，即第一個節點序號為1，第二個節點序號為2，以此類推
        try:
            num = int(input("請輸入需要插入的節點位置序號："))  # 輸入序號必須是大於0的正整數，如果輸入大於最後一個節點的序號則插入到最後一個節點之後
            if num < 1:
                print("輸入必須為大於0的正整數")
                continue
        except ValueError:
            print("輸入有誤")
            continue
        ptr = ergodic(head, num-1)  # 獲取需要插入位置的前一個節點
        insert_data = Player()
        insert_data.name = input("姓名：")
        insert_data.score = input("分數：")
        insert_data.next = ptr.next
        ptr.next = insert_data
        if num == 1:  # 如果插入位置是1的話，那麼head將發生變化
            head = insert_data
 
    elif select == '4':  # 刪除連結串列中任意位置的節點
        try:
            num = int(input("請輸入需要刪除的節點位置序號："))  # 輸入序號必須是大於0的正整數，如果輸入大於最後一個節點的序號則刪除最後一個節點
            if num < 1:
                print("輸入必須為大於0的正整數")
                continue
        except ValueError:
            print("輸入有誤")
            continue
        ptr = ergodic(head, num - 1)  # 獲取需要刪除位置的前一個節點
        if ptr == ergodic(head, num):  # 輸入序號過大時需要做特殊處理，因為輸入序號過大也代表刪除最後一個節點，那麼這時我需要獲取這最後一個節點的前一個節點
            ptr = ergodic(ptr)
        ptr.next = ptr.next.next
        if num == 1:  # 如果刪除位置是1的話，那麼head將發生變化
            head = ptr.next
 
    elif select == '5':  # 反轉連結串列
        new_first = invert(head)  # 獲取新的第一個節點
        head = new_first  # head指向新的第一個節點
        print('成功反轉')
 
    elif select == '6':
        print("成功離開")
        break
    else:
        print("輸入錯誤，請重試")