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("输入错误,请重试")

部分运行结果如下:

发布了49 篇原创文章 · 获赞 10 · 访问量 9250

猜你喜欢

转载自blog.csdn.net/heibuliuqiu_gk/article/details/102650593