提交网站今日冯站长之家
目录
- 一、题目
- 二、解法
- 完整代码
一、题目
给定循环单调非递减列表中的一个点,写一个函数向这个列表中插入一个新元素 insertVal ,使这个列表仍然是循环非降序的。
给定的可以是这个列表中任意一个顶点的指针,并不一定是这个列表中最小元素的指针。
如果有多个满足条件的插入位置,你可以选择任意一个位置插入新的值,插入后整个列表仍然保持有序。
如果列表为空(给定的节点是 null),你需要创建一个循环有序列表并返回这个节点。否则,请返回原先给定的节点。
示例 1:
输入:head = [3,4,1], insertVal = 2
输出:[3,4,1,2]
解释:在上图中,有一个包含三个元素的循环有序列表,你获得值为 3 的节点的指针,我们需要向表中插入元素 2 。新插入的节点应该在 1 和 3 之间,插入之后,整个列表如上图所示,最后返回节点 3 。
示例 2:
输入:head = [], insertVal = 1
输出:[1]
解释:列表为空(给定的节点是 null),创建一个循环有序列表并返回这个节点。
示例 3:
输入:head = [1], insertVal = 0
输出:[1,0]
提示:
0 <= Number of Nodes <= 5 * 104
-106 <= Node.val, insertVal <= 106
二、解法
因为是非递减排序,所以直接遍历链表就好了,找到对应的位置插入。
只不过情况有些多,分情况讨论就ok。
如果找到了,就正常插入。
但是有三种特殊情况:
第一种情况,insertval比链表中所有值都大
第二种情况,insertval比链表中所有值都小
第三种情况,链表中所有值一样
又因为是循环链表,一定要有一个条件,判断已经转过一圈了。
完整代码
"""
# Definition for a Node.
class Node:def __init__(self, val=None, next=None):self.val = valself.next = next
"""class Solution:def insert(self, head: 'Optional[Node]', insertVal: int) -> 'Node':# 找到插入位置a = headif a is None:new = Node(insertVal)new.next = newreturn newpre, cur = head.next, headflag = False# 找到最大值while True:# 正常插入if pre.val >= insertVal >= cur.val:cur.next = Node(insertVal, pre)flag = Truebreak# 特殊情况1、2elif cur.val > pre.val:if insertVal <= pre.val or insertVal >= cur.val:cur.next = Node(insertVal, pre)flag = Truebreakcur, pre = cur.next, pre.nextif cur == head:breakif flag:return head# 特殊情况3cur.next = Node(insertVal, pre)return head