LeetCode剑指 Offer II 029. 排序的循环链表

题目描述

本题目来自LeetCode上的『剑指 Offer II 029. 排序的循环链表』

给定循环单调非递减列表中的一个点，写一个函数向这个列表中插入一个新元素 insertVal ，使这个列表仍然是循环升序的。

给定的可以是这个列表中任意一个顶点的指针，并不一定是这个列表中最小元素的指针。

如果有多个满足条件的插入位置，可以选择任意一个位置插入新的值，插入后整个列表仍然保持有序。

如果列表为空（给定的节点是 null），需要创建一个循环有序列表并返回这个节点。否则。请返回原先给定的节点。

示例1：

输入：head = [3,4,1], insertVal = 2
输出：[3,4,1,2]
解释：在上图中，有一个包含三个元素的循环有序列表，你获得值为 3 的节点的指针，我们需要向表中插入元素 2 。新插入的节点应该在 1 和 3 之间，插入之后，整个列表如上图所示，最后返回节点 3 。

提示

0 <= Number of Nodes <= 5 * 10^4
-10^6 <= Node.val <= 10^6
-10^6 <= insertVal <= 10^6

题解

设新节点为 node，当前节点为 curr，当前节点的下一个结点为 next。分类讨论，一共有五种情况：

空链表：加入 node，并 node->next = node。
只有一个节点：将 node 插到 head 后面，并 node->next = head。

排除上两种情况后，链表可能只有非递减的一段，也有可能是非递减的两段（链表的最大值后连着最小值）：

只有一段：找到 curr->val <= insertVal <= next->val 的位置后插入 node。也有可能原链表所有的值都相等，将 node 插入到链表尾部。
有两段且 insertVal > curr->val：将 node 插入到 curr 之后。
有两段且 insertVal < next->val：将 node 插入到 curr 之后。

代码

class Solution {
public:
    Node* insert(Node* head, int insertVal) {
        Node *node = new Node(insertVal);
        if (head == nullptr) {
            head = node;
            head->next = head;
            return head;
        }
        if (head->next == head) {
            head->next = node;
            node->next = head;
            return head;
        }
        Node *curr = head, *next = head->next;
        while (next != head) {
            if (curr->val <= insertVal && insertVal <= next->val) {
                break;
            }
            if (curr->val > next->val) {
                if (insertVal > curr->val || insertVal < next->val) {
                    break;
                }
            }
            curr = curr->next;
            next = next->next;
        }
        curr->next = node;
        node->next = next;
        return head;
    }
};