题目描述
一个环形单链表从头节点head开始不降序,同时由最后的节点指回头节点。给定这样一个环形单链表的头节点head和一个整数num,请生成节点值为num的新节点,并插入到这个环形链表中,保证调整后的链表依然有序。
struct ListNode {
int val;
ListNode *next;
ListNode() : val(0), next(nullptr) {}
ListNode(int x) : val(x), next(nullptr) {}
ListNode(int x, ListNode *next) : val(x), next(next) {}
};
class Solution {
public:
ListNode* insertNum(ListNode* head, int num) {
}
};
思路
思路:
- 根据 n u m num num生成一个新节点 n o d e node node
- 如果链表为空,让 n o d e node node自己组成一个环形链表,然后返回node
- 如果链表不为空,令变量
p
r
e
=
h
e
a
d
,
c
u
r
=
h
e
a
d
−
>
n
e
x
t
pre = head,cur=head->next
pre=head,cur=head−>next,然后pre和cur同步后移
- 如果 p r e − > v a l < = n u m < = c u r − > v a l pre->val <= num <= cur->val pre−>val<=num<=cur−>val,那么node插入到 p r e pre pre和 c u r cur cur之间,然后返回 h e a d head head
- 如果
p
r
e
和
c
u
r
r
pre和curr
pre和curr转了一圈,也没有发现上面情况,此时有两种可能:要么node节点值比链表中每个值都大,要么node节点值比链表中每个值都要小。此时
n
o
d
e
node
node应该插入到头结点前面
- 举例:1->2->3->1…,num=5,应该把5插入到1前面,此时返回原来头结点1
- 举例:1->2->3->1…,num=0,应该把0插入到1前面,此时将node作为新头节点返回,即返回0
class Solution {
public:
ListNode* insertNum(ListNode* head, int num) {
ListNode *node = new ListNode(num);
if(head == nullptr){
node->next = node;
return node;
}
ListNode *prev = head, *curr = head->next;
while (curr != head){
if(prev->val <= num && curr->val >= num){
break;
}
prev = curr;
curr = curr->next;
}
prev->next = node;
node->next = curr;
return head->val < num ? head : node;
}
};
时间复杂度为O(N),空间复杂度为O(1)