给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
题目给的函数签名为:
public ListNode partition(ListNode head, int x) {
}
把原链表分成两个小链表,一个链表中的元素大小都小于 x,另一个链表中的元素都大于等于 x,最后再把这两条链表接到一起,就得到了题目想要的结果。
具体编程的思路为:
①为两个小链表设置虚拟的头结点;
②设置用于遍历、拓展小链表的指针,指针初始指向虚拟头结点;
③设置用于遍历原链表的指针,放入 while 循环中遍历;
④对于原链表中的每个节点,根据其与所给 x 的大小关系,在相应的小链表中添加与正遍历到的节点数值相同的新节点(与题解做法不同,题解中的做法可以抽象理解为将原链表中的节点移动到小链表中);
⑤当完成一次④中操作后,小链表&原链表 对应的指针都要向后移一位;
⑥重复④⑤直至原链表全部遍历,将元素都大于等于 x的小链表接到元素都小于 x的链表后面即可。
package LinkedList;
/**
* @author: LYZ
* @date: 2022/7/1 16:28
* @description: 给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,
* 使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
* 你应当 保留 两个分区中每个节点的初始相对位置。
*/
public class Partition {
public static void main(String[] args) {
ListNode head = new ListNode(1, new ListNode(4, new ListNode(3, new ListNode(2, new ListNode(5, new ListNode(2))))));
Partition par = new Partition();
ListNode ans = par.partition(head, 3);
while (ans != null) {
System.out.println(ans.val);
ans = ans.next;
}
}
public ListNode partition(ListNode head, int x) {
// 存放小于 x 的链表的虚拟头结点
ListNode dummySmall = new ListNode(-1);
// 存放大于等于 x 的链表的虚拟头结点
ListNode dummyBig = new ListNode(-1);
// 负责生成结果链表的指针
ListNode pointSmall = dummySmall;
ListNode pointBig = dummyBig;
// 负责遍历原链表
ListNode pointHead = head;
while (pointHead != null) {
if (pointHead.val < x) {
pointSmall.next = new ListNode(pointHead.val);
pointSmall = pointSmall.next;
} else {
pointBig.next = new ListNode(pointHead.val);
pointBig = pointBig.next;
}
pointHead = pointHead.next;
}
pointSmall.next = dummyBig.next;
return dummySmall.next;
}
}
可以与 21. 合并两个有序链表 一起分析,体会虚拟头结点存在的意义。
当做题时需创建新链表时,为新链表使用虚拟头结点,并设置遍历该链表的指针。