您的位置:首页 > 其它

[leetcode] 86. Partition List

2015-12-15 16:29 253 查看
Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater
than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,

Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

这道题是按给定值分割链表,小于给定值的在前,不小于给定值的在后,保持原来的相对顺序,题目难度为Medium。

对于任意节点,如果小于给定值则不做处理,直接遍历下一节点,如果不小于给定值则将其取出依次放置到新链表尾部,然后修改它之前节点的next指针指向它的下一个节点,最终将新的链表接在原链表尾部即可。整体比较简单,关键在于链表头的处理上。具体代码:
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* cur = head;
ListNode* lPre = NULL;
ListNode* rPre = NULL;
ListNode* rHead = NULL;
while(cur) {
if(cur->val < x) {
lPre = cur;
}
else {
if(!lPre) head = cur->next;
else lPre->next = cur->next;
if(!rPre) rHead = cur;
else rPre->next = cur;
rPre = cur;
}
cur = cur->next;
}
if(lPre) lPre->next = rHead;
else head = rHead;
if(rPre) rPre->next = NULL;
return head;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode