1. 题目描述
2. 思路分析
整体思路:创建两个链表,分别存放小于x的结点和大于等于x的结点,分别进行尾插。
这道题目使用哨兵位会更简单,原因如下(能避开很多为空的情况):
(1)使用哨兵位就不需要考虑两个链表尾插时为空的情况。
(2)两个链表链接时也不需要考虑是否为空的情况。
(3)哪怕有一个链表为空,也有哨兵位的头结点,正常链接即可。
我们用结构体指针lhead和ltail分别表示值小于x的那一条链表,用结构体指针ghead和gtail表示值大于等于x的那一条链表。
用malloc()函数分别申请两个结点,也就是两个链表的哨兵位,让lhead和ltail一开始指向其中一个,ghead和gtail一开始指向另一个。
再创建一个结构体指针cur用来遍历原链表,我们采用了while循环,当cur不为空时遍历结点。
当结点的值小于x时,我们将这个结点尾插到第一个链表(ltail->next=cur)。再让ltai往后走一步(ltai=ltail->next)。
当结点的值大于等于x时,我们将结点尾插到第二个链表(gtail->next=cur)。再让gtail往后走一 一步(gtail=gtail->next)。
尾插一个结点后,让cur往后走一步(cur=cur->next)。当cur为空时停止循环。
然后将两个链表链接起来。(ltail->next=ghead->next)。
有一点需要非常注意!!!
将gtail->next=NULL
否则可能会出现环。
因为现在lhead指向的是哨兵位,所以我们要将lhead往后走一步(lhead=lhead->next)。
因为怕找不到lhead的下一个位置,所以我们引入一个结构体指针head保存lhead的下一个位置。(struct ListNode *head=lhead->next)。
然后为了防止内存泄漏,我们要用free()释放掉两个哨兵位(即free(lhed)和free(ghead))。
最后返回head即可。
3. 代码实现
/*
struct ListNode {
int val;
struct ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};*/
class Partition {
public:
ListNode* partition(ListNode* pHead, int x) {
struct ListNode *lhead,*ltail,*ghead,*gtail;
lhead=ltail=(struct ListNode*)malloc(sizeof(struct ListNode));
ghead=gtail=(struct ListNode*)malloc(sizeof(struct ListNode));
struct ListNode *cur=pHead;
while(cur)
{
if(cur->val<x)
{
ltail->next=cur;
ltail=ltail->next;
}
else
{
gtail->next=cur;
gtail=gtail->next;
}
cur=cur->next;
}
ltail->next=ghead->next;
//不空,可能导致出现环
gtail->next=NULL;
struct ListNode *head=lhead->next;
free(lhead);
free(ghead);
return head;
}
};