leetcode 203 移除链表元素

        题目要求:给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足

 Node.val == val 的节点,并返回 新的头节点 。

        假设有三个节点A B C ,   链表删除可表示为

代码可表示为 A->next = A->next->next

表示A节点的下一个元素从节点B(A->next) 指向 C(A->next->next)

但是从上述删除结点的例子中,需要三个节点才能完成删除。如果头节点也满足删除条件

head->val == val 就无法实现,所以要先判断头节点。

if(head!=null && head->val == val){

head = head->next;     //把头节点的下一个值赋值给头节点 

}

        显然,使用if判断仍有缺陷 , 因为只能删除一个节点,如果后续仍有节点满足,则无法满足题意。

假设需要删除的val 的值为1 , 下图则需要删除三个节点,需要用while循环遍历后续节点判断是否应该删除。

所以

while(head!=null && head->val == val){

head = head->next;     //把头节点的下一个值赋值给头节点 

}才是正确写法

注意, 将head 赋给一个临时变量: ListNode cur = head;

保证head作为头节点,保留对头部的引用。

完整代码:

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        while(head!=null && head.val == val){
            head = head.next;
        }

        ListNode cur = head;
        while(cur!= null && cur.next!=null){
            if(cur.next.val == val){
                cur.next = cur.next.next;
            }else{
                cur = cur.next;
            }
        }
        return head;
    }
}
  • 8
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值