LeetCode 86 分隔链表

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/jsszwc/article/details/87951248

题目：

https://leetcode-cn.com/problems/partition-list/

题意：

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

思路：

新建两个头节点，依次划分链表中的节点到头节点后面，也就是以 $x$为基准划分成两个链表，最后再合并。

代码：

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode *dummy1 = new ListNode(0);
        ListNode *dummy2 = new ListNode(0);
        ListNode *curNode1 = dummy1, *curNode2 = dummy2;
        while(head) {
            if(head->val < x) {
                curNode1->next = head;
                curNode1 = curNode1->next;
            } else {
                curNode2->next = head;
                curNode2 = curNode2->next;
            }
            head = head->next;
        }
        curNode2->next = nullptr; //这里要置为nullptr，否则最后一个节点小于x时，会造成环状链表
        curNode1->next = dummy2->next;
        return dummy1->next;
    }
};