leetcode -remove linked list elements

题目 remove linked list elements

描述

Remove all elements from a linked list of integers that have value val.
Example:
Input: 1->2->6->3->4->5->6, val = 6
Output: 1->2->3->4->5

解法
c++

链表前加一结点方便直接遍历
然后开始比较删除结点

class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
if(head == NULL)
return NULL;
ListNode FirstNode(0);
FirstNode.next = head;
ListNode* p = &FirstNode;
while(p != NULL)
{
ListNode* tmp = p->next;
if(tmp != NULL && tmp->val == val )
{
p->next = tmp->next;
delete tmp;
}
else
p= p->next;
}
return FirstNode.next;
}
};

---

python

class Solution(object):
def removeElements(self, head, val):
“”"
:type head: ListNode
:type val: int
:rtype: ListNode
“”"
prev = ListNode(-1)
prev.next = head
orig = prev
while head:
if head.val ==val:
prev.next = head.next
else:
prev = head
head = head.next
return orig.next