题目
给定循环单调非递减列表中的一个点,写一个函数向这个列表中插入一个新元素 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 * 10^4
-10^6 <= Node.val <= 10^6
-10^6 <= insertVal <= 10^6
注意:本题与主站 708 题相同: https://leetcode-cn.com/problems/insert-into-a-sorted-circular-linked-list/
解题
方法一:链表 模拟
思路
首先把要插入的节点(insertNode
)新建出来,如果传入 null
的话就让该节点指向自己然后返回。
然后新建一个引用(curr
)指向 head
,因为头节点最后要用于返回,不能拿它来直接进行循环修改。
先找出该循环链表中的最大节点:
- 向后枚举每个节点,如果当前节点比下一个节点大,那么该节点就是该非递减循环链表中的最大节点。
- 如果枚举到当前节点的下一个节点是头节点,那就说明该链表中节点的值全部相同,任何一个节点都是最大节点。
那么找到的最大节点(max
)后一个节点就一定是最小节点(min
)(因为该链表是非递减循环链表)。
如果要插入的值 比最大节点的值大 或者 比最小节点的值小 那么就可以把节点插在最大节点的后面。
否则就从 min
开始枚举链表把节点插到合适的地方。
代码
class Solution {
public Node insert(Node head, int insertVal) {
Node insertNode = new Node(insertVal);
if (head == null) {
insertNode.next = insertNode;
return insertNode;
}
Node curr = head;
while (curr.val <= curr.next.val && curr.next != head) curr = curr.next;
Node max = curr, min = curr.next;
if (insertVal >= max.val || insertVal <= min.val) {
insertNode.next = max.next;
max.next = insertNode;
return head;
}
for (curr = min; curr != max; curr = curr.next) {
if (curr.val <= insertVal && curr.next.val >= insertVal) {
insertNode.next = curr.next;
curr.next = insertNode;
return head;
}
}
return head;
}
}
评论区