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

2020-08-04 17:11:17 浏览数 (1)

算法简介

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

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

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

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

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

代码实现:

代码语言:javascript复制
/**
 * 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
}

执行结果:

0 人点赞