移除单链表元素(Java)

最新推荐文章于 2024-04-21 16:21:25 发布

片云天共远

最新推荐文章于 2024-04-21 16:21:25 发布

阅读量794

点赞数

文章标签：算法 leetcode java 开发语言后端

本文链接：https://blog.csdn.net/weixin_47869348/article/details/123901657

版权

这篇博客详细介绍了如何在不使用虚拟头节点和使用虚拟头节点的情况下，解决LeetCode上的203题——移除链表中的指定元素。代码实现中分别展示了两种情况的处理：链表全空和头节点值等于待删除值。通过这两种解法，可以有效地从单链表中删除特定值的所有节点。

摘要由CSDN通过智能技术生成

leetcode.203

题目要求

单链表的定义

 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; }
 }

不添加虚拟头节点

特殊情况
1.链表全空
2.head.val == val

代码

public ListNode removeElements(ListNode head, int val) {
        if(head == null) return head;//全空，直接返回
        
        ListNode temp = head;
        while(temp.next != null) {
            if(temp.next.val == val) {
                temp.next = temp.next.next;
             } else temp = temp.next;
        }
        if(head.val == val) head = head.next;
        return head;
    }

添加虚拟头节点

如果添加一个虚拟头节点，next指向head，那么以上的两个情况则不需要特殊考虑。

代码

public ListNode removeElements(ListNode head, int val) {
        ListNode newHead = new ListNode(-1, head);

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