编写程序以 x 为基准分割链表,使得所有小于 x 的节点排在大于或等于 x 的节点之前。如果链表中包含 x,x 只需出现在小于 x 的元素之后(如下所示)。分割元素 x 只需处于“右半部分”即可,其不需要被置于左右两部分之间。
示例:
输入: head = 3->5->8->5->10->2->1, x = 5
输出: 3->1->2->10->5->5->8
这几题的题目描述是真的蠢,看都看不懂,大致的意思我是懂的但是这个实例一出来就不是很懂了,我对题目的理解就是比x小的在左边,比x大的在右边,但是他没有说到一个点就是左边和右边的排序问题啊,每个人的答案肯定是不太一样的,应该是有好几种算法来做的,看看实例的输出我也不知道他是什么算法,看了一下评论区的答复好像是随便一种都可以,他也不早说。
我想的是双指针法,一起走,一根去遍历,一根去找最近的大于等于x的值,遍历的时候把小的弄上来,就完成了,但是我自己没有敲,因为我看到了另外的算法就是分类法,小的一类,大的一类,最后拼起来,我觉得这个也不错,就直接借用一下别人的好了
代码如下:(别人敲的)
* 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) {
// 用双指针也可以
// 小于x的在链表左边,大于等于x的放在链表右边
// 设置两个虚拟结点,一个连接小于x的结点,一个连接大于等于x的结点,最后将两个链表连起来即可
ListNode small = new ListNode(); // 虚拟头结点 小于x
ListNode big = new ListNode(); // 虚拟头结点 大于等于x
ListNode cur = head;
ListNode smallCur = small;
ListNode bigCur = big;
while(cur != null){
if(cur.val < x){
smallCur.next = cur;
smallCur = smallCur.next;
}else{
bigCur.next = cur;
bigCur = bigCur.next;
}
cur = cur.next;
}
// 拼接两个链表
smallCur.next = big.next;
bigCur.next = null;
return small.next;
}
}