给定一棵有 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但是如何解决呢?