算法篇:链表之删除和为0的元素

时间:2022-07-22
本文章向大家介绍算法篇:链表之删除和为0的元素,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

算法简介

利用前缀和的方法,例如前缀和[3,5,6,3,7],那么第一个3和最后一个3之间的节点之和就是0,不然的这两个数字不可能相等

// 1.基于上面的原理,我们采用 两次hash的办法,第一次hash找到相同前缀和的最后一个位置的节点
// 2. 第二次hash找到第一次出现和相同的节点位置,那么将这两个节点之间的所有检点都删除,即可

笔者觉得这类题目可以扩展成,多数之和为固定值的场景,不过有个小技巧需要注意,在求和的时候,记得减去这个固定值,如此一来便退化成了多数之和为0的情况。

题目1: 链表中删除综合值为0的连续节点

https://leetcode-cn.com/problems/remove-zero-sum-consecutive-nodes-from-linked-list/

代码实现:

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func removeZeroSumSublists(head *ListNode) *ListNode {
    m1 := make(map[int]*ListNode)
    pre := new(ListNode)
    pre.Next = head
    sum := 0
    for n:=pre; n!=nil; n=n.Next {
        sum += n.Val
        m1[sum] = n // 保存最后一个位置的相同节点的内容
    }
    
    s := 0 
    for m:=pre; m != nil; m=m.Next {
        s += m.Val
        m.Next = m1[s].Next // 删除相同sum第一次出现和最后一次出现之间的节点内容
    }
    return pre.Next
}

执行结果: