边工作边刷题:70天一遍leetcode: day 21
Partition List
要点:这题的循环条件要用cur.next,why?如果用cur,要维护两个pre,比较麻烦,而用cur.next,cur就是pre了。另一个好处是如果cur.next为null,只有一个结点不需要移动任何结点。
还有一个特殊条件是判断lastcur,这样不移动。凡是有边界指针的题,都需要考虑这个(比如insertion sort list)。这里就有问题,为什么不能和不相等的情况用同样的code path呢?因为移动结点的假设是将待移动结点(即cur.next)的next连上last.next,如果lastcur,那么待移动结点和last.next是同一个,这样就接成自环了。
错误点:
- cur的初始点为dummy,而不是head
- 别忘了交换结点之后lasat指针的移动
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def partition(self, head, x):
"""
:type head: ListNode
:type x: int
:rtype: ListNode
"""
dummy = ListNode(0)
dummy.next = head
last = dummy
cur = dummy
while cur and cur.next:
if cur.next.val<x:
if last==cur:
last=last.next
cur=cur.next
else:
next = cur.next.next
cur.next.next = last.next
last.next = cur.next
cur.next = next
last=last.next
else:
cur=cur.next
return dummy.next