目标
给你一个链表的头节点 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
思路
将链表中由零分隔的节点合并为一个节点(将它们的值相加),然后删除零节点。链表头与尾为零,中间节点也可能为零。
使用栈保存非零节点,如果栈非空并且当前节点值为零,则合并栈中节点,并将其值赋值给前面的零节点,然后将最后一个非零节点的 next 赋值给前面零节点的 next。
其实也没必要用栈保存,直接累加到前面零节点的值上即可。
代码
/**
* @date 2024-09-09 9:09
*/
public class MergeNodes2181 {
public ListNode mergeNodes_v2(ListNode head) {
// 用于操作head中的零节点,初始化为空节点,第一次修改并没有实际修改链表中的节点值
ListNode zeroNode = new ListNode();
ListNode res = head;
while (head.next != null) {
if (head.val == 0) {
// 上一个零节点的next指向当前零节点
zeroNode.next = head;
// 指向当前零节点
zeroNode = head;
} else {
zeroNode.val += head.val;
}
head = head.next;
}
zeroNode.next = null;
return res;
}
}
性能








