LeetCode. 86. 分隔链表
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
Related Topics:链表,双指针
思路分析:
要分隔链表,很明显需要两个新的链表来存储分割开的数据,所以new两个节点来生成链表。
此处再生成两个哑节点,方便后期用拼接和输出用。
具体模块:遍历head
,如果小于要求的数,存到小节点的链表里,然后小节点后移,head
后移。如果不是,则存到大节点的链表里,同样节点和head
后移。直到遍历结束。
注意遍历结束要记得让大节点的next
指向null
。
最后一步就是合并两个链表:此时大节点遍历结束,指向的是null。因此,我们让遍历结束的小节点.next
指向最开始生成大节点的哑节点头。就完成了连接
最后输出最开始生成小节点的哑节点.next
。
具体代码:
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode partition(ListNode head, int x) {
ListNode prev1 = new ListNode();
ListNode prev = prev1;
ListNode cur1 = new ListNode();
ListNode cur = cur1;
while (head != null){
if(head.val < x){
prev.next = head;
prev = head;
}
else{
cur.next = head;
cur = head;
}
head = head.next;
}
cur.next = null;
prev.next = cur1.next;
return prev1.next;
}
}
运行: