leetcode 【 Partition List 】python 实现

题目

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

 

代码:oj测试通过 Runtime: 53 ms

 1 # Definition for singly-linked list.
 2 # class ListNode:
 3 #     def __init__(self, x):
 4 #         self.val = x
 5 #         self.next = None
 6 
 7 class Solution:
 8     # @param head, a ListNode
 9     # @param x, an integer
10     # @return a ListNode
11     def partition(self, head, x):
12         if head is None or head.next is None:
13             return head
14         
15         h1 = ListNode(0)
16         dummyh1 = h1
17         h2 = ListNode(0)
18         dummyh2 = h2
19         
20         p = head
21         while p is not None:
22             if p.val < x :
23                 h1.next = p
24                 h1 = h1.next
25             else:
26                 h2.next = p
27                 h2 = h2.next
28             p = p.next
29         
30         h1.next = dummyh2.next
31         h2.next = None

思路

这道题的意思就是:把比x小的单拎出来,把大于等于x的单拎出来,再把两个Linked List首尾接起来。

技巧是设定两个虚拟表头dummyh1和dummyh2:保留住两个新表头。

另外需要设定两个迭代变量h1和h2:用于迭代变量

posted on 2015-01-10 11:06  承续缘  阅读(250)  评论(0编辑  收藏  举报

导航