移除链表元素
给你一个链表的头节点 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也连进去。所以在正确的解法中,在循环中每次新建一个节点,用来存储需要复制的节点。这样保证每次复制进去的都只是一个节点。