LeetCode_数据结构入门_移除链表元素

移除链表元素

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == 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
输出：[]

（上述题目来源于LeetCode）

解法

/**
 * 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) {
        //新建一个哨兵节点,方便后面返回移除元素之后的链表
        ListNode resultHead = new ListNode(-1);
        //定义一个指针节点,指针最开始指向哨兵节点
        ListNode point = resultHead;
        
        //以节点不为空为循环条件进行循环
        while(head != null){
            //如果循环到该节点的值不是需要移除的
            if(head.val != val){
                //在循环中每次新建一个节点,将原链表中要复制的节点复制给node,然后node给point,
                ListNode node = new ListNode(head.val);
                //然后point指针将这个节点放到最后的结果链表中
                point.next = node;
                //指针向后递进
                point = point.next;
            }
            //递进到下一节点
            head = head.next;
        }
        //返回排列后的链表
        return resultHead.next;
    }
}

关于这道题目，我之前写的是

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        ListNode resultHead = new ListNode(-1);
        ListNode point = resultHead;

        while(head != null){
            if(head.val != val){
                point.next = head;
                head = head.next;
                point = point.next;
            }else{
                head = head.next;
            }
        }
        return resultHead.next;
    }
}

但是这样写总是会出现问题，比如上述题目的示例1，就会返回[1,2,3,4,5,6]，就是中间的6它会跳过去，但是最后的6就不会跳过去。如果返回point，就会输出[5,6]。说明此时point指向5时，head指向6，本来是要跳过的然后结束循环的，但是由于是链表，point指向5时会把6也连进去。所以在正确的解法中，在循环中每次新建一个节点，用来存储需要复制的节点。这样保证每次复制进去的都只是一个节点。