题目:跳转至 86. 分隔链表
给你一个链表和一个特定值 x ,请你对链表进行分隔,使得所有小于 x 的节点都出现在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入:head = 1->4->3->2->5->2, x = 3
输出:1->2->2->4->3->5
/**
* Definition for singly-linked list.
* 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* partition(ListNode* head, int x) {
}
};
思路:
没什么特别想说的,这题,难就难在他是个链表。遍历 head 中的值,小于 x 的存到一个链表,大于 x 的存进另一个链表,最后两个链表合并即可。
链表肯定要画图!
具体分割时:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* small = new ListNode(0); //small指向当前链表的末尾节点
ListNode* smallHead = small; //设置一个哑节点,他的next指向链表头节点,以方便处理头节点为空的情况
ListNode* large = new ListNode(0);
ListNode* largeHead = large;
while (head != nullptr) {
if (head->val < x) {
small->next = head;
small = small->next;
} else {
large->next = head;
large = large->next;
}
head = head->next;
}
large->next = nullptr; //置空
small->next = largeHead->next; //将small链表尾节点指向large链表的头节点
return smallHead->next;
}
};