Leetcode刷题笔记 86. 分隔链表

时间：2021年1月3日
知识点：链表拆分
题目链接

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

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

示例：
输入：head = 1->4->3->2->5->2, x = 3
输出：1->2->2->4->3->5

解法：

1. 我们需要自己模拟一遍
2. 可以发现 相当于把链表分隔开 小于特定值的为一链 大于等于的为一链
3. 设置两个头节点 分别对应小的一链 和 大的一链
4. 遍历链表 如果小于特定值 放入小的一链 大于特定值 放入大的一链
5. 最后大小链 合并

代码

//头文件
#include "cheader.h"
struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode *shead = new ListNode(0);
        ListNode *s = shead;
        ListNode *lhead = new ListNode(0);
        ListNode *l = lhead;
        while(head != nullptr){
            if(head->val < x){
                s->next = head;
                s = s->next;
            }else{
                l->next = head;
                l = l->next;
            }
            head = head->next;
        }
        s->next = lhead->next;
        l->next = nullptr;
        return shead->next;
    }
};
int main()
{
    ListNode *head = new ListNode(1);ListNode *n1 = new ListNode(4);ListNode *n2 = new ListNode(3);
    ListNode *n3 = new ListNode(2);ListNode *n4 = new ListNode(5);ListNode *n5 = new ListNode(2);
    head->next = n1; n1->next = n2; n2->next = n3;
    n3->next = n4; n4->next = n5;
    Solution s;
    int x = 3;
    ListNode *h = s.partition(head, x);
    while(h != nullptr){
        cout<<h->val<<endl;
        h = h->next;
    }
    return 0;
}

今天也是爱zz的一天哦！