题目
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
解析
这个题是链表分组的中等难度题,其实没有什么难度,我把题看错了,看成了拓展中的那种题(可以看下拓展)。我就说怎么做都通不过。原题的话,只需要分别建立两个链表就行了,val<x的放在一个链表,>=x的放在一个链表,然后将两个链表连接起来就行了。时间复杂度O(n),空间复杂度是o(1)因为是直接用的现有节点拼凑。下面是我的代码:
```java
public ListNode partition(ListNode head, int x) {
if (head == null || head.next == null){
return head;
}
//小于x的那一条链表
ListNode before = new ListNode(-1);
ListNode beforeCur = before;
//大于等于x的那一条链表
ListNode after = new ListNode(-1);
ListNode afterCur = after;
while (head != null){
if (head.val < x){
beforeCur.next = head;
beforeCur = beforeCur.next;
}else{
afterCur.next = head;
afterCur = afterCur.next;
}
head = head.next;
}
afterCur.next = null;
//最后将两条链表连接起来
beforeCur.next = after.next;
return before.next;
}
拓展
我把这个题看成了小于x的都在前面,大于等于x的都在x的后面,可以有多个x节点,最后要保证顺序除3以外的节点不能改变顺序,当然可以向上面的题那样分成三个链表,也可以不管x节点,直接删掉,统计x节点的个数,最后在添加上去,各位可以动手实现一下。