删除链表的节点
给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。
返回删除后的链表的头节点。
注意:此题对比原题有改动
示例 1:
输入: head = [4,5,1,9], val = 5
输出: [4,1,9]
解释: 给定你链表中值为 5 的第二个节点,那么在调用了你的函数之后,该链表应变为 4 -> 1 -> 9.
示例 2:
输入: head = [4,5,1,9], val = 1
输出: [4,5,9]
解释: 给定你链表中值为 1 的第三个节点,那么在调用了你的函数之后,该链表应变为 4 -> 5 -> 9.
解题思路:
从头到尾尾遍历,遇到链表的值与val相等时,把后面的节点的值赋给当前节点,并将指针指向下下个节点(下下个节点非空的情况),如果下下个节点为空,判断删除倒数第一个还是倒数第二个节点,如果删除倒数第二个:将倒数第一个节点的值赋给倒数第二个,并将倒数二个节点的next设置为空,如果删除倒数第一个,直接将倒数二个节点的next设置为空。
java
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode deleteNode(ListNode head, int val) {
ListNode p = head;
while(p.next.next!=null){
if( p.val == val){
p.val = p.next.val;
p.next = p.next.next;
return head;
}
else{
p = p.next;
}
}
if(p.next.next == null){
if( p.val == val){
p.val = p.next.val;
p.next = null;
return head;
}
if( p.next.val == val){
p.next = null;
return head;
}
}
return head;
}
}
python
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def deleteNode(self, head, val):
"""
:type head: ListNode
:type val: int
:rtype: ListNode
"""
p = head
while p.next.next :
if p.val == val:
p.val = p.next.val
p.next = p.next.next
return head
else:
p = p.next
if p.next.next == None:
if p.val == val:
p.val = p.next.val
p.next = None
return head
if p.next.val == val:
p.next = None
return head
return head