难度中等357
给你一个链表和一个特定值
x
,请你对链表进行分隔,使得所有小于x
的节点都出现在大于或等于x
的节点之前。你应当保留两个分区中每个节点的初始相对位置。
示例:
输入:head = 1->4->3->2->5->2, x = 3 输出:1->2->2->4->3->5
原本想的是用栈来存储大小的数值,后来超时了,。。我
然后瞄了一眼题解,啊明白了!
把小于和大于x的节点分别形成两个子链表,最后对子链进行链接就好啦。不需要声明临时节点存储,直接用节点的插入就好。
/**
* 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) {
ListNode* small = new ListNode(0);
ListNode* smallHead = small;
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;
return smallHead->next;
}
};
果然还是做中等题欢喜更多呢!还是在不断学习中!加油!