[C/C++]数据结构 链表OJ题: 链表分割

题目描述:

        现有一链表的头指针 ListNode* pHead，给一定值x，编写一段代码将所有小于x的结点排在其余结点之前，且不能改变原来的数据顺序，返回重新排列后的链表的头指针。

解题思路:

        创建两个新链表,head1,head2, 遍历原链表,把小于x的结点尾插到head1链表中,把大于等于x的结点尾插到head2链表,最后链接head1和head2,这个题代码实现起来很简单,但是要注意的是几种特殊情况,比如分类后head1为空或head2为空等情况,还需要用if语句单独判断,包括代码中使用结点的next,都需要提前判断一下该结点是否为空,所以我们可以给这两个新链表加上哨兵位,这样就不需要head1,head2就永远不会为空,这样就可以省去很多单独判断语句

简单介绍一下哨兵位:

       单链表分为带头结点和不带头结点两种情况。带头结点的第一个节点叫哨兵位，属于附加的链表节点，不保存有效数值，只储存第一个有效节点的地址，负责找到第一个节点。

代码实现:

class Partition {
public:
    ListNode* partition(ListNode* pHead, int x) {
         //带哨兵位
         ListNode* head1,*tail1,*head2,*tail2;
         head1=tail1=(ListNode*)malloc(sizeof(ListNode));
         head2=tail2=(ListNode*)malloc(sizeof(ListNode));

         ListNode* cur=pHead;
        //小于x的结点尾插到head1,大于x的结点尾插到head2
         while(cur)
         {
            if(cur->val<x)
            {
                tail1->next=cur;
                tail1=tail1->next;
            }
            else 
            {
                tail2->next=cur;
                tail2=tail2->next;
            }

            cur=cur->next;
         }
         //链接两个链表
         tail1->next=head2->next;
         tail2->next=NULL;
         pHead=head1->next;

         free(head1);
         free(head2);
         
         return pHead;
         }
};