LeetCode刷题——分隔链表#86#Medium

分隔链表的思路探讨与源码
组合的题目如下图，该题属于链表和双指针类型的题目，主要考察对于遍历方法的理解和链表特性的使用。本文的题目作者想到2种方法，分别是递归方法和双指针方法，其中递归方法使用Java进行编写，而双指针方法使用Python进行编写，当然这可能不是最优的解法，还希望各位大佬给出更快的算法。
在这里插入图片描述
本人认为该题目可以使用递归方法解决，首先判断链表是否为空或者只有一个元素，如果是则直接返回。然后开始递归循环调用，判断传入的值如果比当前值要大并且比下一个值要小，则将head放到所有比它小的节点之后，否则就直接返回结果，最终结束递归循环。那么按照这个思路我们的Java代码如下：

#喷火龙与水箭龟
class Solution {
    public ListNode partition(ListNode head, int x) {
        if(head == null || head.next == null){
            return head;
        }
        head.next = partition(head.next, x);
        if(head.val >= x && head.next.val < x){
            ListNode index = head;
            ListNode nextNode = head.next;
            while(index.next != null && index.next.val < x){
                index = index.next;
            }
            head.next = index.next;
            index.next = head;
            return nextNode;
        }else{
            return head;
        }
    }
}

在这里插入图片描述
显然，我们看到递归方法的效果不错，同时还可以使用双指针的方法进行处理。首先判断是否为空，并且初始化参数，将链表的头部作为左指针的初始位置，开始遍历链表，找到第一个比目标值x要大的前一个位置，然后把右指针记为大于等于x的位置开始遍历，直到链表尾端，通过这种方式遍历后将最终结果返回。所以按照这个思路就可以解决，下面是Python代码部分：

#喷火龙与水箭龟
class Solution:
    def partition(self, head: ListNode, x: int) -> ListNode:
        if not head:
            return 
        index=ListNode(0)
        index.next=head
        smallPart=index
        while smallPart.next and smallPart.next.val<x:
                smallPart=smallPart.next 
        bigPart=smallPart.next 
        while bigPart:                        
            while bigPart.next and bigPart.next.val>=x:
                bigPart=bigPart.next
            if not bigPart.next:
                break
            goat=bigPart.next
            bigPart.next=goat.next  
            goat.next=smallPart.next
            smallPart.next=goat   
            smallPart = goat
        return index.next

在这里插入图片描述
从结果来说Java版本的递归方法的效率不错，而Python版本的双指针方法的速度一般，但应该是有更多的方法可以进一步提速的，希望朋友们能够多多指教，非常感谢。

LeetCode刷题——分隔链表#86#Medium

おすすめ