天天看点

移除链表元素(无傀儡节点,递归+迭代)问题思路代码

移除链表元素

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

问题

一个单链表(无傀儡节点/头结点),删除指定的所有元素。

如 :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(栈溢出问题)。

继续阅读