描述
给出一个长度为n的单链表和一个值,返回一个链表的头节点,要求新链表中小于x的节点全部在大于等于x的节点左侧,并且两部分之内的节点与原来的链表保持相对顺序不变。
例:
输入:{1,4,3,2,5,2},3
返回:{1,2,2,4,3,5}
要求:时间O(n),空间O(1)
解题思路
创建两个链表分别指向大于x和小于x
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(!head||!head->next) return head;
ListNode *dummy1=new ListNode(0);
ListNode *dummy2=new ListNode(0);
ListNode* p = dummy1;
ListNode* q = dummy2;
while(head){
if(head->val<x){
p->next = head;
p = head;
}else{
q->next = head;
q = head;
}
head = head->next;
}
p->next = dummy2->next;
q->next = nullptr;
return dummy1->next;
}
};
补充
链表对数组和矢量的优点
尽管链表的编码和管理比数组更复杂,但它们有一些明显的优势。首先,链表可以容易地扩大或缩小。实际上,程序员并不需要知道链表中有多少个结点。它们只是根据需要在内存中创建。
有人可能会争辩说,链表并不优于矢量(可在标准模板库中找到),因为它们也可以扩展或缩小。然而,链表对于矢量的优势是结点可以插入链表或从链表中删除的速度。要将值插入矢量的中间,需要将插入点之后的所有元素朝矢量的末尾移动一个位置,从而为新值腾出空间。同样,从矢量中删除一个值需要将删除点之后的所有元素都朝矢量的开始方向移动一个位置。而当一个结点插入链表或从链表中删除结点时,其他结点都不必移动。
ListNode *head = nullptr;
// Create first node with 12.5
head = new ListNode; // Allocate new node
head->value = 12.5; // Store the value
head->next = nullptr; // Signify end of list