LeetCode：203 移除链表元素

持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第5天，点击查看活动详情

题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

示例 1：

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

示例 2：

输入：head = [], val = 1
输出：[]

示例 3：

输入：head = [7,7,7,7], val = 7
输出：[]

提示：

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

解题

解题一：迭代

思路

1. 首先判断头节点是否为目标，如果是目标节点则进行遍历，直到找到第一个非目标节点的值作为头节点
2. 遍历链表，移除指定的目标内容：将前一个元素的 next 指针指向目标指针的 next 指针所指向的位置
3. 返回头节点

代码

/**
 * 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 Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode result = head;
        // 确定头节点
        while (result != null && result.val == val) {
            result = result.next;
        }
        if (result == null) {
            return result;
        }
        ListNode temp = result;
        while (temp!=null && temp.next != null) {
            while (temp.next != null && temp.next.val == val) {
                temp.next = temp.next.next;
            }
            temp = temp.next;
        }
        return result;
    }
}

总结

性能分析

• 执行耗时：1 ms，击败了 66.87% 的 Java 用户
• 内存消耗：42.5 MB，击败了 5.04% 的 Java 用户

解题二：递归

思路

1. 判断节点是否为空，为空则直接返回
2. 节点不为空，通过递归，移除后面节点中的目标节点
3. 设置当前节点的 next 为优化后的节点
4. 判断当前节点是否为目标节点，如果是则返回 nex 节点，否则返回当前节点

代码

/**
 * 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 Solution {
    public ListNode removeElements(ListNode head, int val) {
        if (head != null) {
            head.next = removeElements(head.next, val);
            head = head.val == val ? head.next : head;
        }
        return head;
    }
}

总结

性能分析

• 执行耗时：1 ms，击败了 50.43% 的 Java 用户
• 内存消耗：41.9 MB，击败了 89.80% 的 Java 用户