引入
给定循环单调非递减列表中的一个点,写一个函数向这个列表中插入一个新元素 insertVal
,使这个列表仍然是循环升序的。
给定的可以是这个列表中任意一个顶点的指针,并不一定是这个列表中最小元素的指针。
如果有多个满足条件的插入位置,可以选择任意一个位置插入新的值,插入后整个列表仍然保持有序。
如果列表为空(给定的节点是 null),需要创建一个循环有序列表并返回这个节点。否则。请返回原先给定的节点。
示例
输入:head = [3,4,1], insertVal = 2
输出:[3,4,1,2]
解释:在上图中,有一个包含三个元素的循环有序列表,你获得值为 3 的节点的指针,我们需要向表中插入元素 2 。新插入的节点应该在 1 和 3 之间,插入之后,整个列表如上图所示,最后返回节点 3 。
解法(C++)
这题是一个任意一个顶点的指针,所以第一个思路就是往下遍历,如果找到一个节点cur
满足cur->val <= insertVal <= cur->next->val
,就可以直接插入了;
如果找不到,就说明所有值都比它小或大,所以这个都会插入到边界位置上。
满足insertVal <= cur->next->val <=cur->val
(插入值比最小的小)
或者insertVal >= cur->val >= cur->next->val
(插入值比最大的大)
/*
// Definition for a Node.
class Node {
public:
int val;
Node* next;
Node() {}
Node(int _val) {
val = _val;
next = NULL;
}
Node(int _val, Node* _next) {
val = _val;
next = _next;
}
};
*/
class Solution {
public:
Node* insert(Node* head, int insertVal) {
//判空
if(head == nullptr){
head = new Node(insertVal);
head->next = head ;
return head ;
}
auto cur = head;
while(cur->next !=head){
//如果下一个节点小于当前节点,证明达到了边界位置(末边界就是最大值)
if(cur->next->val < cur->val){
//如果下一个节点大于插入值,证明插入值比最小的小
if(cur->next->val >= insertVal) break;
// 如果当前节点小于插入值,证明插入值比最大值的大
else if(cur->val <= insertVal) break;
}
//如果当前节点小于插入值且下一节点大于插入值,则插入到这个位置
if(cur->val <= insertVal && cur->next->val >= insertVal)
break;
//不满足三个条件则找下一个
cur = cur->next;
}
cur->next = new Node(insertVal,cur->next);
return head;
}
};