1. 问题描述:
给你一个链表的头节点 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/merge-nodes-in-between-zeros/
2. 思路分析:
分析题目可以知道我们可以从链表的头结点开始遍历,判断当前遍历的节点的值是否是0,如果是0说明两段为0的链表的节点到此结束,我们可以声明一个遍历s来统计需要合并链表节点总和,声明一个合并节点之后的链表头结点,指针q指向当前的节点,当发现当前枚举节点的值为0那么创建一个新的链表节点接在q的后面,指针q向后移动,s置为0,最终返回合并链表的头结点即可。
3. 代码如下:
from typing import Optional
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def mergeNodes(self, head: Optional[ListNode]) -> Optional[ListNode]:
s = 0
# res为合并链表的头结点
p, res = head.next, ListNode(-1)
# q为指向当前的合并之后链表的节点
q = res
while p:
s += p.val
# 判断当前的节点是否为0, 为0说明上一段不是0的链表就结束了, 此时需要新建节点并且
# 接在合并之后链表的节点后面
if p.val == 0:
q.next = ListNode(s)
s = 0
q = q.next
p = p.next
# 返回下一个节点
return res.next