LeetCode86-分隔链表

昨天刷微博突然看到了一首情诗特别有意思

是胡适先生在《梦与诗》一书中所写的

内容如下：

醉过才知酒浓

爱过才知情重

你不能做我的诗

正如我不能做你的梦

---

86-分隔链表

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

思路：

碰到这种解链表的题目，正如我之前经常说的那样，可以先把链表中每个节点的值保存到一数组中，再对该数组进行处理，这样比较方便。本题亦是如此，解题步骤大致如下：

1. 将头指针每个节点的值存储到一列表中
2. 分别用两个列表来保存小于x以及不小于x的值
3. 最后将两个列表整合起来即可

代码如下：

class Solution(object):
    # 本题的解题步骤如下：1. 将头指针每个节点的值存储到一列表中
    # 2. 分别用两个列表来保存小于x以及不小于x的值
    # 3. 最后将两个列表整合起来即可
    def partition(self, head, x):
        """
        :type head: ListNode
        :type x: int
        :rtype: ListNode
        """
        # 将头指针每个节点的值存储到一列表中
        head_copy = head
        head_list = []
        while head_copy:
            head_list.append(head_copy.val)
            head_copy = head_copy.next
        # 分别定义左右列表来保存小于x以及不小于x的值
        left_list = []
        right_list = []
        for index in range(len(head_list)):
            if head_list[index] < x:
                left_list.append(head_list[index])
            else:
                right_list.append(head_list[index])
        return left_list+right_list


if __name__ == "__main__":
    head_list = [1, 4, 3, 2, 5, 2]
    head = ListNode(0)
    head_copy = head
    for index in head_list:
        head_copy.next = ListNode(index)
        head_copy = head_copy.next
    x = 3
    new_head = Solution().partition(head.next, x)
    print(new_head)

不过执行效率着实有些低，只有可怜的4%左右，悲催！！！