leetcode No203 移除链表元素 java

Posted 2021-07-11 短腿Cat

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了leetcode No203 移除链表元素 java相关的知识，希望对你有一定的参考价值。

题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回新的头节点。
在这里插入图片描述

题解

迭代写法

简单题简单做，我们循环一遍，找到则删掉该节点，考虑到单链表特殊性（因为单链表不可以找前一个元素），我们就让两个挨在一块儿的节点一起往后移动，让后一个节点值和val比较，如果节点值等于val，我们可以利用它的前一个节点来删除它自己，代码如下：

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) return null;
        for (ListNode a = head, b = a.next;b != null; ) {
            if (b.val == val && b.next != null) {
                a.next = b.next;
            } else if(b.val == val && b.next == null) {
                a.next = null;
                break;
            } else a = a.next;
            b = a.next;
        }
        if (head.val == val) return head.next;
        else return head;
    }
}

递归写法：

根据链表的定义，我们可以递归定义它，它拥有递归性，因此链表题目常可以用递归的方法求解。可以用递归实现：

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) {
            return head;
        }
        head.next = removeElements(head.next, val);
        return head.val == val ? head.next : head;
    }
}
作者：LeetCode-Solution