203. Remove Linked List Elements

Remove all elements from a linked list of integers that have value val.

Example

Given: 1 –> 2 –> 6 –> 3 –> 4 –> 5 –> 6, val = 6

Return: 1 –> 2 –> 3 –> 4 –> 5

Credits:
Special thanks to @mithmatt for adding this problem and creating all test cases.

题目不难，但是经历了一个代码渐渐优化的过程

删除一个链表元素，重要的是记录删除位置的前驱和后继，因此设置两个指针专门保存待删除元素的前驱和后继，加上待删除元素一共三个指针，程序如下：

class Solution(object):
def removeElements(self, head, val):
"""
:type head: ListNode
:type val: int
:rtype: ListNode
"""
if not head:
return None

p = head
q = p.next
pre = ListNode(-1)
pre.next = p
h = pre
while p != None:
if p.val == val:
pre.next = q
else:
pre = pre.next
p = p.next
if p != None:
q = p.next

return h.next

然后AC了，but 看下图时间开销比较大啊，



才击败了25.92%的人 T_T，要优化优化才行！

发现其实只要两个指针就可以了，一个保存待删除元素的前驱，一个保存待删除元素，它的后继用next就能得到，我之前傻了啊。

改后程序如下，比原来少了一个pre指针：

class Solution(object):
def removeElements(self, head, val):
"""
:type head: ListNode
:type val: int
:rtype: ListNode
"""
if not head:
return None
h = ListNode(-1)
h.next = head
p = h
q = p.next

while q != None:
if q.val == val:
p.next = q.next
else:
p = p.next
q = p.next
return h.next

时间上快了一点，击败了62%的人，然后继续优化，发现不用加上这个判断：

if not head:
return None

因为以前做题习惯了，总喜欢加上一个判断，避免输入为空的情况，但是这里不需要加，因为while循环位置有判断了。所以果断删除改后代码如下：

class Solution(object):
def removeElements(self, head, val):
"""
:type head: ListNode
:type val: int
:rtype: ListNode
"""
#         if not head: # 没有这句了
#             return None
h = ListNode(-1)
h.next = head
p = h
q = p.next

while q != None:
if q.val == val:
p.next = q.next
else:
p = p.next
q = p.next
return h.next

击败了90%的人，此时无声胜有声啊，话说那些dalao们是怎么写的程序。

总结：写完程序以后尝试优化优化，不要因为题简单AC就行了，有时候强迫自己一下，或许会有提高呢