算法day3:链表

203. 移除链表元素


给你一个链表的头节点 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. 应该要确定怎么删除元素 

              

①. (中间节点)假设我们要删除的元素是cur.next,我们只需要将cur.next指向cur.next.next即可,而不是先让cur.next = null,这样我们就找不到cur.next.next了。

②. (头节点)我们只需要将head设置为head.next即可,就完成了头节点的删除。

2. 所以现在只需要做两件事

①. 判断头节点是否和target相同,如果相同,删除头节点。

②. 判断中间节点是否和target相同,如果相同,删除中间节点。

代码实现

//直接在原来的链表上删除
class Solution {
    public ListNode removeElements(ListNode head, int val) {
        while(head != null && head.val == val) {
            head = head.next;
        }
        ListNode curNode = head;
        while(curNode != null && curNode.next != null) {
            if(curNode.next.val == val) {
                curNode.next = curNode.next.next;
            } else {
                curNode = curNode.next;
            }
        }
        return head;
    }
}

方法二:设置一个虚拟头节点指向原有链表的头节点

好处

好处就是不需要区分删除头节点和中间节点,此时的头节点也是中间节点,我们只需要按照删除中间节点的方法删除即可。

代码实现

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode privNode = new ListNode(0, head);
        ListNode curNode = privNode;
        while(curNode.next != null) {
            if(curNode.next.val == val) {
                curNode.next = curNode.next.next;
            } else {
                curNode = curNode.next;
            }
        }
        return privNode.next;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值