题目描述
思路:创建两个新的链表,把小于x的节点尾插到第一个链表,把大于等于x的节点尾插到第二个链表,然后把第二个链表链接在第一个链表后面
题目要求返回重新排列后的链表的头指针,正常情况下肯定是要返回第一个链表的头节点,但是我们尝试分析一下特殊情况:
若第一个链表为空呢?就不需要链接了,而且应该返回第二个链表的头节点,因此要考率特殊情况的话就很麻烦
所以这道题我们选择使用带哨兵位的头节点来解决这个问题,这样的话无论第一个链表为不为空,我们都返回的是第一个链表的带哨兵位的头节点
具体过程示意如下
现在我们要分割这个链表,假定x是4
定义两个链表中的带哨兵位的头节点,尾节点(方便尾插)
分割过程
-----------------------------------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------------------------------
-----------------------------------------------------------------------------------------------------------------------------
这时两个链表均已创建完成,下面需要把第二个链表链接到第一个链表后面
lesstail->next = greaterhead->next
主体部分就结束啦,还有一些细节大家注意~
①链接两个链表的是链接的是第一个链表的尾部和第二个链表的头部(不是带哨兵位的头节点)
②上述过程完成后需要把greatertail->next 置成空指针(链表的尾节点的next就应该是空指针)
③最后返回新的头节点,返回的是greatertail->next,而不是带哨兵位的头节点(lesstail)
④养成好的习惯,最后释放掉两个带哨兵位的头节点
这里大家理解一下上文提到过的定义哨兵位的头节点的优势~~~
当所有的节点都比指定的节点大的时候,这时候第一个链表为空,只有带哨兵位的头节点,我们返回lesstail->next 的时候,也就返回了第二个链表的头节点,因此避免了讨论
代码如下
这是牛客网的上的题目,没有C语言的选择,因此默认用c++,c++是向下兼容c语言的,所以照常写C语言代码即可~
class Partition {
public:
ListNode* partition(ListNode* pHead, int x) {
struct ListNode* lesstail, *lesshead, *greatertail, *greaterhead;
greatertail = greaterhead = (struct ListNode*)malloc(sizeof(struct ListNode));
lesstail = lesshead = (struct ListNode* )malloc(sizeof(struct ListNode));
struct ListNode* cur = pHead;
while(cur)
{
if(cur->val < x)
{
lesstail->next = cur;
lesstail = lesstail->next;
}
else
{
greatertail->next = cur;
greatertail = greatertail->next;
}
cur = cur->next;
}
lesstail->next = greaterhead->next;
greatertail->next = NULL;
struct ListNode* Phead = lesshead->next;
free(greaterhead);
free(lesshead);
return Phead;
}
};