【LintCode 简单】96. 链表划分

最新推荐文章于 2020-07-19 22:30:59 发布

原创最新推荐文章于 2020-07-19 22:30:59 发布 · 221 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LintCode #Python #单链表 #归并排序

LintCode 专栏收录该内容

90 篇文章

订阅专栏

本文介绍了一种链表处理算法，该算法能够将一个给定的单链表中所有节点按给定数值x进行划分，使得所有小于x的节点排在大于等于x的节点前面，同时保持各部分内部节点原有顺序不变。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.问题描述：

给定一个单链表和数值x，划分链表使得所有小于x的节点排在大于等于x的节点之前。

你应该保留两部分内链表节点原有的相对顺序。

2.样例：

给定链表 1->4->3->2->5->2->null，并且 x=3

返回 1->2->2->4->3->5->null

3.代码：

第一种方法：

"""
Definition of ListNode
class ListNode(object):

    def __init__(self, val, next=None):
        self.val = val
        self.next = next
"""


class Solution:
    """
    @param: head: The first node of linked list
    @param: x: An integer
    @return: A ListNode
    """
    def partition(self, head, x):
        # write your code here
        if head is None:  
            return head  
        # 新建两个节点  
        first = ListNode(-1)  
        second = ListNode(-1)  
        cur1, cur2 = first, second  
        # 遍历原始链表  
        while head:  
            if head.val < x:  
                cur1.next = head  
                head = head.next  
                cur1 = cur1.next  
            else:  
                cur2.next = head  
                head = head.next  
                cur2 = cur2.next  
        # 合并  
        cur1.next = second.next  
        cur2.next = None  
        return first.next

第二种方法：

"""
Definition of ListNode
class ListNode(object):

    def __init__(self, val, next=None):
        self.val = val
        self.next = next
"""


class Solution:
    """
    @param: head: The first node of linked list
    @param: x: An integer
    @return: A ListNode
    """
    def partition(self, head, x):
        # write your code here
        if head is None:
            return None
        elif head.next is None:
            return head
        else:
            tail=head
            count=1
            while tail.next:
                tail=tail.next
                count+=1
            pre=ListNode(0)
            cur=head
            pre.next=cur
            i=0
            fg=True
            while i<count:
                if cur.val>=x:
                    tail.next=cur
                    pre.next=cur.next
                    tail=cur
                    if fg:
                        head=head.next
                    tail.next=None
                    cur=pre.next
                else:
                    fg=False
                    pre=pre.next
                    cur=cur.next
                i+=1
            return head