给你一个链表的头节点 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
输出:[]
提示:
- 列表中的节点数目在范围
[0, 104]
内 1 <= Node.val <= 50
0 <= val <= 50
题解:
线性遍历法:
// 时间复杂度:O(n)
// 空间复杂度:O(1)
func removeElements(head *ListNode, val int) *ListNode {
// 创建一个虚拟头节点(哨兵节点)
dummyHead := &ListNode{}
// 哨兵节点成为链表的头节点,后面的遍历可以直接从这个哨兵节点开始
dummyHead.Next = head
cur := dummyHead
// 遍历链表
for cur != nil && cur.Next != nil {
if cur.Next.Val == val {
// 如果下个节点值等于 val,则将当前节点指向到下下个节点,直接跳过下个节点
cur.Next = cur.Next.Next
} else {
// 如果下个节点的值不等于 val,则将当前节点改为下个节点,继续判断
cur = cur.Next
}
}
return dummyHead.Next
}
递归法:
// 时间复杂度:O(n)
// 空间复杂度:O(n)
func removeElements(head *ListNode, val int) *ListNode {
if head == nil{
return nil
}
head.Next = removeElements(head.Next, val)
if head.Val == val {
return head.Next
}
return head
}
评论前必须登录!
注册