2181. 合并零之间的节点

https://leetcode.cn/problems/merge-nodes-in-between-zeros/description/?envType=daily-question&envId=2024-09-09

题目描述

给你一个链表的头节点 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

解题思路

[[Excalidraw/2181合并零之间的节点]]

代码编写

public ListNode mergeNodes(ListNode head) {  
    int tempSum = 0;  
    ListNode result = new ListNode();  
    // 结果listNode的指向游标  
    ListNode resultNext = result;  
    // 遍历的指向游标  
    ListNode next = head;  
    while (next != null) {  
        if (next.val == 0) {  
            if (resultNext.next == null) {  
                resultNext.next = new ListNode();  
            } else {  
                resultNext.next.val = tempSum;  
                resultNext = resultNext.next;  
                resultNext.next = new ListNode();  
                tempSum = 0;  
            }  
        } else {  
            tempSum += next.val;  
        }  
        next = next.next;  
    }  
    // 移除最后的0  
    resultNext.next = null;  
    return result.next;  
}

单元测试

System.out.println(new Solution2181().mergeNodes(TransformUtils.arrays2ListNode("[0,3,1,0,4,5,2,0]")));

额外补充