给定一棵有 n 个节点的树 T,T 中有多少个子树 (T') 最多有 K 条边连接到 (T – T')? 输入格式
第一行包含两个整数 n 和 K,后面跟着 n-1 行,每行包含两个整数 a 和 b,表示 a 和 b 之间存在边。
约束
1 <= K <= n <= 50 每个节点都用从 1 到 n 的不同数字表示。
输出格式
一个整数,表示可能的子树的数量。
def cutTree(n, k, edges):
g = [[] for _ in range(n)]
for i in edges:
g[i[0]-1].append(i[1]-1)
g[i[1]-1].append(i[0]-1)
global ans
ans = 1
def multiply(x, y):
ans = defaultdict(lambda: 0)
for k, v in x.items():
for k1, v1 in y.items(): ans[k+k1-1] += v*v1
for k, v in ans.items():
if k in x: x[k] += v
else: x[k] = v
def dfs(i,p):
global ans
if g[i] == [p]:
ans += 1
return {0:1}
x = 1 if i else 0
res = {len(g[i])-x : 1}
for nxt in g[i]:
if nxt != p: multiply(res, dfs(nxt, i))
ans += sum(((v if i+x <= k else 0) for i, v in res.items()))
return res
dfs(0,-1)
return ans
它解决了https://www.hackerrank.com/challenges/cuttree/forum但是如何解决呢?
这是我可以给您的解释(如果您需要澄清,请告诉我)。
大意
由于 T 的子树T′是一组连通的节点。
想象一下从某个节点开始扩展这样的子树,然后针对每个相邻节点(DFS 树中的子节点)决定是否要附加它。
如果不附加分支,则该边将被切断。
观察结果
动态规划
代码使用动态规划来有效地计算不同分支的数量(是的,尽管 Hackerrank 论坛/讨论成员声称如此,但此代码使用的是底层 dp)。
目前,代码创建了一个字典 dp 表,其中的键是子树(包括当前节点)具有的切边数量,值是实现该切边的方法数量。
请注意,“乘法”函数结合了 dp 状态。例如:
a)。假设您有一个当前配置,其中包含 k(edgeCount1)个边界边(来自当前节点的部分子树)和一个可以以产生 k1(edgeCount2)个边界边的方式连接的子子树。(再次注意我们正在保存 1 个切口的观察结果)
2. 对边界边的可能数量的卷积可以被认为类似于将两个多项式(其中指数跟踪边界边的数量)相乘,移位为 -1。
是什么让这段代码变得棘手
作者让代码变得如此难以理解的原因之一是他们使用了名称糟糕的变量,在某些情况下,这些名称确实分散了人们对主要思想的注意力。因此,我提供了更新的代码(它简化了部分逻辑,但也以一种更容易理解代码的方式命名了变量):
我的解释可能不够简洁(就简化而言),我鼓励其他人发布自己的答案来解释代码。
希望我的示例代码和调整后的变量名能够稍微解释一下代码的作用
(不过我想指出的是,我确实删除了 dfs 中的基线情况,因为没有必要……就解释而言,这可能是一个错误(因为可能更容易看到其他情况,或者您也可以尝试理解这一点)……
我再次鼓励其他人发布他们的解释,如果您不理解其中的一部分,请随时发表评论。