【python/M/86】Partition List

最新推荐文章于 2021-02-13 20:19:31 发布

alicelmx

最新推荐文章于 2021-02-13 20:19:31 发布

阅读量221

点赞数

分类专栏： LeetCode 文章标签： leetcdeo

本文链接：https://blog.csdn.net/alicelmx/article/details/81331550

版权

LeetCode 专栏收录该内容

79 篇文章 0 订阅

订阅专栏

题目

这里写图片描述

基本思路

简单题
解释下题目：就是把小于x的节点放入x之前，大于x的节点不要管他
创建两个头节点，然后遍历初始链表，若节点值小于x，就放入新链表1，反之放入节点2，最后将两个新链表连接起来即可。

实现代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def partition(self, head, x):
        """
        :type head: ListNode
        :type x: int
        :rtype: ListNode
        """
        # 将大于等于x的结点取出来组成一个新链表，然后将旧的接在新的后面即可
        dummy1,dummy2 = ListNode(0),ListNode(0)
        p1,p2 = dummy1,dummy2
        # 遍历指针
        p = head

        while p:
            if p.val >= x:
                p2.next = p
                p = p.next
                p2 = p2.next
                p2.next = None

            else:
                p1.next = p
                p = p.next
                p1 = p1.next
                p2.next = None

        p1.next = dummy2.next
        return dummy1.next