力扣（LeetCode)每日一题 2181. 合并零之间的节点

最新推荐文章于 2024-09-11 17:16:29 发布

一般路过糸.

最新推荐文章于 2024-09-11 17:16:29 发布

阅读量349

点赞数 5

文章标签：算法 leetcode java

本文链接：https://blog.csdn.net/Jilit_jilit/article/details/142093633

版权

题目链接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

我的代码：

创建一个新节点res。遍历一遍。

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode mergeNodes(ListNode head) {
        ListNode h = head;
        ListNode res = new ListNode();
        ListNode resIdx = res;
        int ns = 0;
        while(h != null){
            if(h.val == 0){
                if(ns != 0){
                    resIdx.next = new ListNode(ns);
                    resIdx = resIdx.next;
                    ns = 0;
                }
            }
            if(h.val != 0){
                ns += h.val;
            }
            h=h.next;
        }
        return res.next;
    }
}