LeetCode Remove Linked List Elements

Description:

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

Solution:

需要单独考虑链表头。

import java.util.*;

public class Solution {
	public ListNode removeElements(ListNode head, int val) {
		while (head != null && head.val == val) {
			head = head.next;
		}

		if (head == null)
			return null;

		ListNode temp = head;
		while (temp.next != null) {
			if (temp.next.val == val) {
				temp.next = temp.next.next;
			} else {
				temp = temp.next;
			}
		}

		return head;
	}
}