LeetCode--2181. 链表--合并零之间的节点

合并零之间的节点

问题描述

给你一个链表的头节点 head，该链表包含由 0 分隔开的一连串整数。链表的 开端 和 末尾 的节点都满足 Node.val == 0。

对于每两个相邻的 0，请你将它们之间的所有节点合并成一个节点，其值是所有已合并节点的值之和。然后将所有 0 移除，修改后的链表不应该含有任何 0。

返回修改后链表的头节点 head。

示例

示例 1

输入：head = [0,3,1,0,4,5,2,0]
输出：[4,11]
解释：

• 标记为绿色的节点之和：3 + 1 = 4
• 标记为红色的节点之和：4 + 5 + 2 = 11

示例 2

输入：head = [0,1,0,3,0,2,2,0]
输出：[1,3,4]
解释：

• 标记为绿色的节点之和：1 = 1
• 标记为红色的节点之和：3 = 3
• 标记为黄色的节点之和：2 + 2 = 4

提示

• 列表中的节点数目在范围 [3, 2 * 10^5] 内
• 0 <= Node.val <= 1000
• 不存在连续两个 Node.val == 0 的节点
• 链表的 开端 和 末尾 节点都满足 Node.val == 0

优化前

func mergeNodes(head *ListNode) *ListNode {
	p := head
	res := &ListNode{}
	l := res
	sum := 0
	for p != nil  {
		if p.Val == 0&&p!=head{
			add := &ListNode{}
			add.Val = sum
			l.Next = add
			l = add
			sum = 0
		
		} else {
			sum += p.Val
		}
		p = p.Next
	}
	return res.Next
}

优化后

func mergeNodes(head *ListNode) *ListNode {
    dummy := &ListNode{} // 创建哑结点以简化结果链表的操作
    current := dummy     // 用于构建新链表
    sum := 0             // 用于累加非零节点的值

    for node := head.Next; node != nil; node = node.Next { // 从head.Next开始，跳过第一个零节点
        if node.Val == 0 { // 遇到零节点
            current.Next = &ListNode{Val: sum} // 将累加值作为新节点加入结果链表
            current = current.Next             // 移动到新节点
            sum = 0                            // 重置累加值
        } else {
            sum += node.Val // 累加非零节点的值
        }
    }

    return dummy.Next // 返回结果链表的头节点
}