1. 程式人生 > >LeetCode_203. 移除連結串列元素

LeetCode_203. 移除連結串列元素

public class S_203 {
    public ListNode removeElements(ListNode head, int val) {
        // 為空
        if (head == null)
            return null;
        // 當不為空 且連結串列值等於給定值時
        while (head != null && head.val == val) {
            // 位移一位
            head = head.next;
        }

        ListNode node = head;

        while (node.next != null) {
            if (node.next.val == val)
                node.next = node.next.next;
            else
                node = node.next;
        }
        return head;
    }
}