16 ms solution for https://leetcode.com/problems/remove-linked-list-elements/description/

linked_list_remove_element_base.go

// solves https://leetcode.com/problems/remove-linked-list-elements/description/
// runtime is 16ms

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */

func removeElements(head *ListNode, val int) *ListNode {
    if head == nil {
        return nil
    }
    for head != nil && head.Val == val {
        head = head.Next
    }
    if head == nil {
        return nil
    }
    p := head
    for p.Next != nil {
        if p.Next.Val == val {
            p.Next = p.Next.Next
        } else {
            p = p.Next
        }
    }
    return head
}