4.LeetCode.86. Partition List【链表划分】

最新推荐文章于 2024-09-18 15:03:01 发布

一个人的世界中

最新推荐文章于 2024-09-18 15:03:01 发布

阅读量87

点赞数

分类专栏： LeetCode 文章标签：链表指针单链表 leetcode

本文链接：https://blog.csdn.net/q1345191/article/details/107929393

版权

LeetCode 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

本文详细讲解了LeetCode中第86题的解决方案，涉及如何将单链表划分为两个连续子链表的问题，重点讨论了使用指针操作链表的方法。

摘要由CSDN通过智能技术生成

链表划分【86】(medium)

#include <iostream>
using namespace std;

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
  public:
    ListNode *partition(ListNode *head, int x) {
        ListNode *less_head = new ListNode(-1); // 临时头节点
        ListNode *more_head = new ListNode(-1);
        ListNode *less_ptr = less_head; // 指针指向头节点
        ListNode *more_ptr = more_head;

        while (head) {
            if (head->val < x) { // 节点值小于x, 将该节点插入less_ptr后
                less_ptr->next = head;
                less_ptr = head;
            } else { // 否则该节点插入more_ptr后
                more_ptr->next = head;
                more_ptr = head; // 链表完成后, less_ptr后移,指向head
            }
            head = head->next;
        }
        less_ptr->next = more_head->next; // less链表尾与more链表头相连
        more_ptr->next = NULL;            // 链表尾结点next置空
        return less_head->next;
    }
};
int main() {
    ListNode a(1);
    ListNode b(4);
    ListNode c(3);
    ListNode d(2);
    ListNode e(5);
    ListNode f(2);
    a.next = &b;
    b.next = &c;
    c.next = &d;
    d.next = &e;
    e.next = &f;

    /**
     * Input: 1->4->3->2->5->2, x = 3
      Output: 1->2->2->4->3->5
     *
     *
     */

    Solution solve;
    ListNode *head = solve.partition(&a, 3);
    while (head) {
        cout << head->val << endl;
        head = head->next;
    }
    return 0;
}