题目:删除链表中等于给定值 val 的所有节点。
示例:
输入: 1->2->6->3->4->5->6, val = 6
输出: 1->2->3->4->5
初始代码:
/**
* 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) {
}
}
解题思路:
第一步:思考初始化代码
我们先看一下题目给的初始代码并思考一下每个值是什么意思
/**
* 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出现的位置
我们先思考这道题是要删除val的所有节点,这时候我们考虑到val所出现的所有位置。链表是否为空?当val出现在链表头时?当val出现在链表中?
第三步:实现当val出现在链表头的逻辑
//这里使用循环,因为你删除头元素,下一个头元素也可能是val
//判断链表头是否为空,链表头的值是否为vval
while(head!=null && head.val==val){
//创建一个要删除的节点 为表头
ListNode removeNode =head;
//让下一个节点重新等于头结点
head=head.next;
//让节点从链表中断开
removeNode.next=null;
}
第三步:实现当val出现在链表中的逻辑
//判断链表是否为空
if(head==null){
return null;
}
//创建一个节点指向下一个节点
ListNode prev=head;
//循环遍历整个链表
while(prev.next!=null){
//如果prev的下一个节点的值是val
if(prev.next.val==val){
//创建要删除的节点
ListNode removeNode =prev.next;
//跳过removeNode节点,指向下一个节点
prev.next=removeNode.next;
removeNode.next=null;
}else{
prev=prev.next;
}
}
第四步:返回head
return head;
整个代码逻辑
class Solution { //头节点 要删除的节点
public ListNode removeElements(ListNode head, int val) {
//这里使用循环,因为你删除头元素,下一个头元素也可能是val
//判断链表头是否为空,链表头的值是否为vval
while(head!=null && head.val==val){
//创建一个要删除的节点 为表头
ListNode removeNode =head;
//让下一个节点重新等于头结点
head=head.next;
//让节点从链表中断开
removeNode.next=null;
}
//判断链表是否为空
if(head==null){
return null;
}
//创建一个节点指向下一个节点
ListNode prev=head;
//循环遍历整个链表
while(prev.next!=null){
//如果prev的下一个节点的值是val
if(prev.next.val==val){
//创建要删除的节点
ListNode removeNode =prev.next;
//跳过removeNode节点,指向下一个节点
prev.next=removeNode.next;
removeNode.next=null;
}else{
prev=prev.next;
}
}
return head;
}
}
提交代码
毫无疑问,我们的代码是正确的。