LeetCode203-移除链表元素

题目描述 给你一个链表的头节点 head 和一个整数 val ,请你删除链表中所有满足 Node.val == val 的节点,并返回 新的头节点 。 示例一 输入:head = [1,2,6,3,4,5,6], val = 6 输出:[1,2,3,4,5]

示例二: 输入:head = [], val = 1 输出:[]

示例三: 输入:head = [7,7,7,7], val = 7 输出:[]

思路一:

  1. 对于 头节点 和 非头节点,需要分别处理
  2. 当为头节点时,进行判断,若为val值,则直接删除头节点
  3. 当为非头节点时,进行判断,若为val值,则让该节点的前一个结点指向该结点的下一个结点;
  4. 因此,我们需要两个变量来存储 当前结点 和 当前结点的前一个结点;

具体实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution{
          
   
	public ListNode removeElements(ListNode head, int val) {
          
   
		//将结点分为头节点和非头节点,需要分别处理
		//为头节点时
		while(head!=null && head.val==val){
          
   
			head = head.next;
		}
		if(head == null){
          
   
			return head;
		}
		//为 非头节点时
			//创建两个变量来存储 当前结点 和 当前结点的前一个结点
		ListNode pre = head;
		ListNode cur = head.next;
			//把cur作为循环条件是因为 cur在每次循环的最后都是有修改的
		while(cur != null){
          
   
			//若当前结点为val值,则把前一个结点指向当前结点的下一个结点
			if(cur.val == val){
          
   
				pre.next = cur.next;
			//若当前结点不为val值,则正常指向下一个结点,也即head结点的移动
			}else{
          
   
				pre = cur;
			}
			//每次循环最后都要进行 循环条件的递增,用于退出循环
			cur = cur.next;
		}
		//最后返回head
		return head;
	}
}

思路二:

  1. 创建一个虚拟结点作为头节点,那么链表中的所有结点的处理操作都是一致的; 实现:
public ListNode removeElements(ListNode head, int val) {
          
   
        if(head == null){
          
   
            return head;
        }
        //创建一个虚拟头节点
        ListNode virtual = new ListNode(-1,head);
        //因此链表中除了头节点外所有结点的操作都是一样的
        ListNode pre = virtual;
        ListNode cur = virtual.next;
        while(cur != null){
          
   
            if(cur.val==val){
          
   
                pre.next = cur.next;
            }else{
          
   
                pre = cur;
            }
            cur = cur.next;
        }
        return virtual.next;
    }
经验分享 程序员 微信小程序 职场和发展