移除链表元素


文章目录

  • 移除链表元素
  • 力扣203题目
  • 首先知道如何写出java中的节点类的构造方法
  • 两种解决方法
  • 总结


力扣203题目

题意：删除链表中等于给定值 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
输出：[]

首先知道如何写出java中的节点类的构造方法

public class ListNode{
	int val;
	ListNode(){}
	ListNode(int val){this.val = val;}
	ListNode(int val,ListNode next){this.val = val;this.next = next;}
}

两种解决方法

法一：设置两个指针，初始化前一个节点pre就是head和当前的节点cur为head的下一个节点

（1）head的值为val的时候，即要移除head的节点，直接将head指针指向head的下一个节点

（2）循环条件：cur节点指向空的时候，指针已经从头走到尾

（3）循环过程：①节点指向的值为val，前一个节点（pre）就指向下下一个节点（cur.next）

②如果不是所需的值，pre节点就直接指向下一个节点（cur）

③判断结束后就把cur移向下一个节点

刚刚看到题目的迷惑：为什么修改过程没有涉及head的指针，而返回头指针的地址，可以返回修改后的链表？

（自己的想法）：设置pre和head相等，就相当pre和head都指向链表的首地址，因此通过删除节点修改链表，都是修改同一个链表

class Solution{
	public ListNode removeElements(List Node,int val){
		//头指针不是空
		while(head != null&&head.val == val){
            head = head.next;
        }
        //头指针是空且不是所需的值就直接返回
        if(head == null){
            return head;
        }
        //设置当前节点pre，cur为下一个节点，也就是pre和cur是前后两个节点
        ListNode pre = head;
        ListNode cur = head.next;
        //头指针不是空，且不是当前值
        while(cur != null){
            if(cur.val == val){
                //本来pre的下一个节点应该指向cur，但是因为要删除这个节点，
                //因此就把pre的下一个节点指向cur的下一个节点 
                //相当于把cur这个节点删除
                pre.next = cur.next;
            }else{
                pre = cur;
            }
            //更新cur移向下一个节点
            cur = cur.next;
        }
        return head;
	}
}

法二：设置一个虚拟头节点，它的下一个节点指向head，好处就是删除头节点的时候方便，直接让虚拟节点直接指向head的下一个节点。

class Solution{
	public ListNode removeElements(ListNode head,int val){ 
		ListNode pre = new ListNode(-1 , head);
        if(head == null){
            return head;
        }
        ListNode cur = head;
        while(cur!=null){
            if(cur.val == val){
                pre.next = head.next; 
            }else{
                pre = cur;
            }
            cur = cur.next;
        }
        return pre.next;
    }
}

总结

设置虚拟节点就只要分两种情况讨论，即

（1）头指针是空；

（2）头指针不为空，可以合并要删除头指针或其他指针。