LCR 029. 循环有序列表的插入(中等)

https://leetcode.cn/problems/4ueAj6/

给定循环单调非递减列表中的一个点,写一个函数向这个列表中插入一个新元素 insertVal ,使这个列表仍然是循环升序的。

给定的可以是这个列表中任意一个顶点的指针,并不一定是这个列表中最小元素的指针。

如果有多个满足条件的插入位置,可以选择任意一个位置插入新的值,插入后整个列表仍然保持有序。

如果列表为空(给定的节点是 null),需要创建一个循环有序列表并返回这个节点。否则。请返回原先给定的节点。

示例 1:
LCR 029. 循环有序列表的插入(中等)_第1张图片
输入:head = [3,4,1], insertVal = 2
输出:[3,4,1,2]
解释:在上图中,有一个包含三个元素的循环有序列表,你获得值为 3 的节点的指针,我们需要向表中插入元素 2 。新插入的节点应该在 1 和 3 之间,插入之后,整个列表如上图所示,最后返回节点 3 。

"""
# Definition for a Node.
class Node:
    def __init__(self, val=None, next=None):
        self.val = val
        self.next = next
"""

class Solution:
    def insert(self, head: 'Node', insertVal: int) -> 'Node':
        # 该题的难点是情况的枚举
        # 情况1:列表为空
        # 情况2:列表仅有1个元素
        # 情况3:列表元素个数超过1
            # ① 列表最小值在最左侧,最大值在最右侧
            # ② 列表最大值和最小值在列表中,非边界出现
        node = Node(insertVal)
        
        # 情况1,直接插入
        if not head:
            node.next = node
            return node
        
        # 情况2,插入到其后面
        if head.next == head:
            head.next = node
            node.next = head
            return head
        
        # 情况3,需要遍历,找到插入位置
        p1, p2 = head, head.next
        while p2 != head:  # 注意遍历条件,如果相等,说明转了1圈了
            if p1.val <= insertVal <= p2.val:
                break
            if p1.val > p2.val and (insertVal >= p1.val or insertVal <= p2.val):
                break
            p1 = p1.next
            p2 = p2.next
        p1.next = node
        node.next = p2
        return head

你可能感兴趣的:(链表,链表,数据结构,算法,python,leetcode)