LeetCode刷题_96.分隔链表_用C++实现

LeetCode刷题_96.分隔链表_用C++实现

解题思路

额外申请两个链表,一个放大数,一个放小数,然后小数链表链上大数链表即可
(有点像归并排序)

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        //设置两个虚拟头结点
        //rsmall存放小于特定值x的一组数
        //rbig存放大于特定值x的一组数
        //p1用来指向rsmall链表的末尾节点
        //p2用来指向rbig链表的末尾节点
        //p用来判断循环何时停止
        //q是工具人
		ListNode rsmall,rbig,*p1=&rsmall,*p2=&rbig,*p=head,*q;//rsmall和rbig相当于老板,p1和p2给他们打工
        while(p){
            q=p->next;//将p->next先存起来,这样就保证头结点和他的next断开时,后边的节点不会消失
            //p->val是当前节点的值
            if(p->val<x){
                //如果当前节点值<x,
                //当前节点就链到p1的next
                //还是那句话,左边是橄榄枝,右边是人才本身
                //接下来的两句代码是经典的马氏三角杀
                p->next=p1->next;
                p1->next=p;//此时将p节点链到p1最后面一位
                //保证p1时刻指向rsmall链表的末尾节点
                p1=p;
            }else{
                //如果当前节点值>x,
                //当前节点就链到p2的next
                //还是那句话,左边是橄榄枝,右边是人才本身
                //接下来的两句代码是经典的马氏三角杀
                p->next=p2->next;
                p2->next=p;
                //保证p2时刻指向rbig链表的末尾节点
                p2=p;
            }
            //在某轮循环中,走到这里,p就指向了rs或rm其中一个链表的最后一个位置
            p=q;//每循环完一轮,就将p指向原链表的下一个链表(注:不用担心,p走后,p指向的那个节点会不会消失,因为:在p指向那个节点的同时,p1或p2也指向了那个节点)
        }
        //此时得到rs和rb两个节点
        //让p1链上rbig的next即可(注:rbig是存储>x的一组链表的头结点)
        p1->next=rbig.next;
        //将rsamll排除即可,得到除rsmall节点外的其他所有节点
        return rsmall.next;
    }
};