难度等级:中等
上一篇算法:
力扣此题地址:
1.题目:分割链表
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你不需要 保留 每个分区中各节点的初始相对位置。
2.解题思路:
1.创建两个链表small 和 large, 再创建两个头结点smallHead 和 largeHead,两个头结点分别指向两个链表的头结点,即small 和 large,便于之后return 到链表第一个结点。
2.然后开始遍历链表,将链表中小于x值的结点连接到small链表中,大于等于x值的结点连接到large链表中,遍历完链表之后,将 small 链表的尾结点指向 large 的头结点即可。
注意:遍历完链表之后,要将 large 的尾结点设置为null,因为两个链表都是连接的原始链表,所以后面可能还会连接着其他的结点。
3.最后两个链表连接在一起之后,返回smallHead.next。smallHead.next也就是链表的第一个真正意义上的结点。
参考:
3.代码实现:
/**
* 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) {
if(head == null){
return head;
}
ListNode small = new ListNode(0);
ListNode smallHead = small;
ListNode large = new ListNode(0);
ListNode largeHead = large;
while (head != null) {
if (head.val < x) {
small.next = head;
small = small.next;
} else {
large.next = head;
large = large.next;
}
head = head.next;
}
large.next = null;
small.next = largeHead.next;
return smallHead.next;
}
}