237 删除链表中的节点 简单
发布者:admin 发表于:442天前 阅读数:494 评论:0

237. 删除链表中的节点 简单

请编写一个函数,使其可以删除某个链表中给定的(非末尾)节点。传入函数的唯一参数为 要被删除的节点 。

现有一个链表 – head = [4,5,1,9],它可以表示为:

示例 1:

输入:head = [4,5,1,9], node = 5
输出:[4,1,9]
解释:给定你链表中值为 5 的第二个节点,那么在调用了你的函数之后,该链表应变为 4 -> 1 -> 9.

示例 2:

输入:head = [4,5,1,9], node = 1
输出:[4,5,9]
解释:给定你链表中值为 1 的第三个节点,那么在调用了你的函数之后,该链表应变为 4 -> 5 -> 9.

提示:

链表至少包含两个节点。

链表中所有节点的值都是唯一的。

给定的节点为非末尾节点并且一定是链表中的一个有效节点。

不要从你的函数中返回任何结果。

代码参考:

package main

import "fmt"

func main() {
    fmt.Println(deleteNode(newList([]int{4, 5, 1, 9}), 5))
    fmt.Println(deleteNode(newList([]int{4, 5, 1, 9}), 9))
    fmt.Println(deleteNode(newList([]int{1, 2}), 1))
}

// 注意 break 条件即可
func deleteNode(head *ListNode, v int) *ListNode {
    if head == nil || head.Next == nil {
        return nil
    }
    pre, cur := head, head.Next
    for {
        if cur.Next == nil || cur.Val == v {
            break
        }
        pre = pre.Next
        cur = cur.Next
    }
    pre.Next = cur.Next
    return head
}