【LeetCode】86、分隔链表

题目描述

给你一个链表的头节点 head 和一个特定值 x ，请你对链表进行分隔，使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。

你应当 保留 两个分区中每个节点的初始相对位置。

示例1

示例2

题解

我们需要构造两个链表higher和less，less链表用来按顺序存储所有小于x的节点，higher链表按顺序存储所有大于等于x的节点。最后将less链表尾节点指向higher链表首节点即可。
为此，我们使higher_head和less_head分别为两个空节点（即其值val没有任何意义），它们的next结点分别指向第一个有意义的结点，这样做更方便处理头节点为空的边界条件。而higher_end和less_end指向分别指向链表的末尾结点。初始时两个均为less=head=空结点。然后开始遍历，判断当前链表的节点值是否小于 x，如果小于就将 less_end 的 next 指针指向该节点，否则将 higher_end 的 next 指针指向该节点。
遍历结束后，需要将higher_end的next指针置空，因为之前这个next可能指向其他错误结点。最后将less_end的next指向higher_head（自身为空结点）的 next。

代码

/**
 * 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){
			ListNode* higher_head=new ListNode(0);
			ListNode* higher_end=higher_head;
			ListNode* less_end = new ListNode(0);
			ListNode* less_head = less_end;
			while(head!=NULL){
				if(head->val<x){
					less_end->next=head;
					less_end=head;
				}else{
					higher_end->next=head;
					higher_end=head;
				}
				head=head->next;
			}
			higher_end->next=nullptr; 
			less_end->next=higher_head->next;
			return less_head->next; 
		}
};