203. 移除链表元素

转载自Leet Code

题目描述

删除链表中等于给定值 val 的所有节点。

示例:

输入: 1->2->6->3->4->5->6, val = 6 输出: 1->2->3->4->5

提示：

• 列表中的节点数目在范围 [0, 104] 内
• 1 <= Node.val <= 50
• 0 <= val <= 50

---

我的代码

\(T(N) = O(N)\), \(S(N) = O(1)\)

{.line-numbers}

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class MySolution203 
{
    public ListNode removeElements(ListNode head, int val) 
    {
    	while (head!=null&&head.val==val)
    		head = head.next;
    	if (head==null) return head;
    	
    	ListNode prev = head; ListNode p = head.next;
    	while (p!=null)
    	{
    		if (p.val==val)
    		{
    			prev.next = p.next;
    			p = p.next;
    		}
    		else 
    		{
    			prev = p;
    			p = p.next;
    		}
    	}
    	return head;
    }
}