《程序员面试金典》--分割链表

题目描述:

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或等于x的结点之前，给定一个链表的头指针 ListNode* pHead，请返回重新排列后的链表的头指针。注意：分割以后保持原来的数据顺序不变。

题目分析：

要是链表换作数组要特别小心，因为搬迁数组要涉及到移动元素的开销，而移动链表的元素则容易很多，我们不需要移动和交换链表中的元素，只是改变一下每个节点的next指针就可以，我们可以创建两个链表，一个链表是存放小于x的元素，另一个链表存放大于等于x的元素，然后遍历一遍原有链表，将小于x元素的节点加到第一个链表上，将大于等于x元素的节点加到第二个链表上。便可实现分割。

程序代码如下：

struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
class Partition {
public:
ListNode* partition(ListNode* pHead, int x) {
// write code here
ListNode* beforeStart=NULL;
ListNode* beforeEnd=NULL;
ListNode* afterstart=NULL,*afterEnd=NULL;
ListNode* headval;
while(pHead)
{
headval=pHead->next;
if(pHead->val<x)
{
if(beforeEnd==NULL)
beforeEnd=beforeStart=pHead;
else
{
beforeEnd->next=pHead;
beforeEnd=pHead;
}
}
else
{
if(afterstart==NULL)
{
afterstart=afterEnd=pHead;
}
else
{
afterEnd->next=pHead;
afterEnd=pHead;
}
}
pHead=headval;
}
if(afterEnd)
afterEnd->next=NULL;
if(beforeEnd!=NULL)
beforeEnd->next=afterstart;
else
beforeStart=afterstart;
return beforeStart;
}
};