剑指offer刷题(十四)

52 阅读1分钟

删除链表的节点

给定单向链表的头指针和一个要删除的节点的值,定义一个函数删除该节点。

返回删除后的链表的头节点。

注意:此题对比原题有改动

示例 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