From c44d011bcfe988751e3dc67c384e388e9c54c7e0 Mon Sep 17 00:00:00 2001 From: Azureki Date: Thu, 13 Sep 2018 12:39:22 +0800 Subject: [PATCH] Remove Linked List Elements --- lc203.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 lc203.py diff --git a/lc203.py b/lc203.py new file mode 100644 index 0000000..7930839 --- /dev/null +++ b/lc203.py @@ -0,0 +1,35 @@ +# Definition for singly-linked list. +# class ListNode(object): +# def __init__(self, x): +# self.val = x +# self.next = None + + +class Solution(object): + def removeElements(self, head, val): + """ + :type head: ListNode + :type val: int + :rtype: ListNode + """ + # q = p = ListNode(0) + # p.next = head + p = head + if head is None: + return None + + while p.next: + tem = p.next + if tem.val == val: + p.next = tem.next + else: + p = p.next + + # return head + # head也有可能被删 + # return q.next + + # 这样也行 + if head.val == val: + head = head.next + return head