Leetcode 2872. Maximum Number of K-Divisible Components

  • [Leetcode 2872. Maximum Number of K-Divisible Components](#Leetcode 2872. Maximum Number of K-Divisible Components)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题思路上我觉得还是挺直接的,就是一个自底向上的团聚算法。

首先,由于题目中保证了所有节点的总和一定是k的倍数,因此事实上我们的任意切割都不会导致剩余其他点的总和无法被k整除。故一个简单直接的思路就是自底向上地进行团聚。

我们首先找到所有的叶子节点,即度为1的点,然后考察他们的value是否能够被k整除,如果可以,那么进行切割即可,这样的切割一定不会影响其他节点。

然后,对于无法进行切割的节点,我们将其合并至其父节点当中,即将两者绑定为一个节点,此时再考察其父节点是否变成了一个叶子节点。

我们只需要重复上述操作直到一直处理到最终的根节点即可得到最终的切分方式。

需要注意的是,由于在切割发生时子节点的和被k整除,因此我们再合并过程中事实上并不需要区分切割或者没有切割,因为后者事实上不会对结果产生影响,这样就进一步简化了我们的代码,虽然逻辑上多绕了一下......

2. 代码实现

给出最终的python代码实现如下:

python 复制代码
class Solution:
    def maxKDivisibleComponents(self, n: int, edges: List[List[int]], values: List[int], k: int) -> int:
        if n == 1:
            return 1
        graph = defaultdict(set)
        for u, v in edges:
            graph[u].add(v)
            graph[v].add(u)
            
        q = [u for u in graph if len(graph[u]) == 1]
        res = 0
        
        while q:
            u = q.pop(0)
            if values[u] % k == 0:
                res += 1
            for v in graph[u]:
                graph[v].remove(u)
                if len(graph[v]) == 1:
                    q.append(v)
                values[v] += values[u]
        return res

提交代码评测得到:耗时1464ms,占用内存34.9MB。

相关推荐
Swift社区5 小时前
LeetCode - #139 单词拆分
算法·leetcode·职场和发展
Dong雨6 小时前
力扣hot100-->栈/单调栈
算法·leetcode·职场和发展
trueEve7 小时前
SQL,力扣题目1369,获取最近第二次的活动
算法·leetcode·职场和发展
九圣残炎9 小时前
【从零开始的LeetCode-算法】3354. 使数组元素等于零
java·算法·leetcode
程序猿小柒10 小时前
leetcode hot100【LeetCode 4.寻找两个正序数组的中位数】java实现
java·算法·leetcode
_OLi_11 小时前
力扣 LeetCode 106. 从中序与后序遍历序列构造二叉树(Day9:二叉树)
数据结构·算法·leetcode
我明天再来学Web渗透11 小时前
【SQL50】day 2
开发语言·数据结构·leetcode·面试
小叶lr12 小时前
idea 配置 leetcode插件 代码模版
java·leetcode·intellij-idea
理论最高的吻16 小时前
98. 验证二叉搜索树【 力扣(LeetCode) 】
数据结构·c++·算法·leetcode·职场和发展·二叉树·c
沈小农学编程16 小时前
【LeetCode面试150】——202快乐数
c++·python·算法·leetcode·面试·职场和发展