题目来源
题目描述
给定循环升序列表中的一个点,写一个函数向这个列表中插入一个新元素,使这个列表仍然是循环升序的。给定的可以是这个列表中任意一个顶点的指针,并不一定是这个列表中最小元素的指针。
如果有多个满足条件的插入位置,你可以选择任意一个位置插入新的值,插入后整个列表仍然保持有序。
如果列表为空(给定的节点是 null),你需要创建一个循环有序列表并返回这个点。否则。请返回原先给定的节点。
下面的例子可以帮你更好的理解这个问题:
在上图中,有一个包含三个元素的循环有序列表,你获得值为 3 的节点的指针,我们需要向表中插入元素 2。
新插入的节点应该在 1 和 3 之间,插入之后,整个列表如上图所示,最后返回节点 3。
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) {
}
};
题目解析
思路
对于insertVal,一共有三种情况:
- 如果head为NULL,那么需要:新建一个节点,然后next指向自己
- 如果head不为NULL,思考,val能插入到哪里?
- 如果能够找到a <= insertVal <= b,那么将insertVal生成的node插入到aNode的后面,bNode的节点
- 如果没有找到这样的区间呢?找出最大值的最后一个点,直接在该最大值的点的后面插入结点即可(因为此时,插入值是最大值或最小值)
class Solution {
public:
Node* insert(Node* head, int insertVal) {
if(head == nullptr){
head = new Node(insertVal);
head->next = head;
return head;
}
Node *curr = head;
Node *big = head; //保存最大值的结点
while (true){
if(insertVal >= curr->val && insertVal <= curr->next->val){
Node *newNode = new Node(insertVal);
newNode->next = curr->next;
curr->next = newNode;
return head;
}
if(curr->val > big->val){
big = curr;
}
if(curr->next == curr){
break;
}
curr = curr->next;
}
Node *newNode = new Node(insertVal);
newNode->next = big->next;
big->next = newNode;
return head;
}
};