移除链表元素（无傀儡节点，递归+迭代）

Posted 2022-12-13 小写丶H

移除链表元素

• 问题
• 思路
• 代码
• • • 迭代
    • 递归

问题

一个单链表（无傀儡节点/头结点），删除指定的所有元素。
如 ：1->3->5->7->5->9 删除5之后；变为：1->3->7->9

思路

可以用递归，但是空间复杂度高。
所以使用迭代。遍历一次链表便删除所有指定节点。
使用两个两个指针 cur 和 prve 。

基本删除思路如图片所示。

还要考虑一些细节，没有判断头结点是否删除。所有写到代码注释里面了。

代码

如果代码没懂，多画图理解一下就会明白。

迭代

public ListNode removeElements(ListNode head, int val) 
        //链表头节点为空直接返回空
        if(head==null)
            return null;
        
        //创建两个指针
        ListNode cur=head.next;
        ListNode prve=head;
        //写进循环里，挑出条件为cur为null时。
        while(cur!=null)
            //判断是否删除
            if(cur.val==val)
                prve.next=cur.next;
                cur=cur.next;
            else
                prve=prve.next;
                cur=cur.next;
            
        
        //注意这里，用来判断头结点是否要删除的，很多人容易忘掉！！！
        if(head.val==val)
            head=head.next;
        
        return head;
    

如果只是删除第一次出现的元素，只需要把循环去掉微改一下就可以。

递归

public ListNode removeElements(ListNode head, int val) 
        if (head == null) 
            return head;
        
        //这里就是关键，需要画图理解。
        head.next = removeElements(head.next, val);
//        if(head.val==val)
//        	return head.next;
//        else
//        	return head;
//        
		//简便写法
        return head.val == val ? head.next : head;
    

递归代码很简洁，但是（较抽象）需要理解，但是如果链表很长，那么不建议使用，防止StackOverflowError(栈溢出问题)。