leetcode笔记_链表的遍历与删除（虚拟头结点的使用）

203. 移除链表元素

删除链表中等于给定值 val 的所有节点。

示例:
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5

思路：链表的增加和删除操作，一定要通过待处理节点的前一个节点来实现，为了解决头结点没有前继节点的特殊性，需要给链表添加一个虚拟头结点。新建指针一个用于索引的指针prev, 指向待处理节点的前继节点，则待处理节点为prev.next. 初始化prev则指向链表的虚拟头结点。

解法1. 不使用虚拟头结点，需要考虑头结点的特殊处理，比较复杂

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode removeElements(ListNode head, int val) {
        
        //无虚拟头结点，需要对头结点做特殊处理
        //将链表开头连续的值为val的节点都删除
        while(head!=null && head.val == val){
            head=head.next;
        }
        
        if(head==null){
            return null;
        }
        
        //对头结点的值不为val的链表继续做删除操作
        ListNode cur = head;
        
        while(cur.next != null){
            if(cur.next.val == val){
                cur.next = cur.next.next;
            }else{
                cur = cur.next;
            }
        }
        
        return head;
        
    }
}

解法2. 使用虚拟头结点，代码简介，思路清晰

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode removeElements(ListNode head, int val) {
        //建立虚拟头结点
        ListNode dummyHead = new ListNode(-1);
        dummyHead.next = head;
        
        //建立遍历链表所用的指针，prev指向的是带处理元素的前一个元素，待处理元素为prev.next
        ListNode prev = dummyHead;
        
        //对链表进行遍历，注意链表为空的判断
        while(prev.next!=null){
            if(prev.next.val == val){
                prev.next = prev.next.next;
            }else{
                prev = prev.next;
            }
        }
        
        return dummyHead.next;               
    }
}

leetcode笔记_链表的遍历与删除（虚拟头结点的使用）

203. 移除链表元素

猜你喜欢