中等

问题描述

给你一个链表的头节点 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 * 105] 内
• 0 <= Node.val <= 1000
• 不 存在连续两个 Node.val == 0 的节点
• 链表的 开端 和 末尾 节点都满足 Node.val == 0

一次遍历

遍历链表，若当前节点元素值不为 $0$，则将面非 $0$ 元素相加到当前节点，直到遇到 $0$ 或者 $null$，合并结束后，若当前节点不为空，则必然是值为 $0$ 的节点，将当前节点指向下一节点的下一节点，重复步骤直到遍历结束。

/**
 * Definition for singly-linked list.
 * function ListNode(val, next) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.next = (next===undefined ? null : next)
 * }
 */
/**
 * @param {ListNode} head
 * @return {ListNode}
 */
var mergeNodes = function (head) {
  let curr = head.next
  while (curr) {
    let next = null
    if (curr.val !== 0) {
      next = curr.next
      while (next && next.val !== 0) {
        curr.val += next.val
        next = next.next
      }
    }
    curr.next = next ? next.next : null
    curr = curr.next
  }
  return head.next
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$