关注
n_m_d = "10 2 2"
f_node = "5 3"
tree = "1 1 1 2 3 3 4 5 5"
n,m,d = [int(x) for x in n_m_d.split()]
tree_gp = {x:set() for x in range(1,n+1)}
tree_node = [int(x) for x in tree.split()]
for i,node in enumerate(tree_node,2):
tree_gp[node].add(i)
tree_gp[i].add(node)
f_node = [int(x) for x in f_node.split()]
def bfs(tree_gp,node,d):
'''
:param tree_gp:tree_map
:param d: max layer
:return: list layer < d
'''
res = set()
queue = []
queue.append(node)
seen = set()
cnt = 0
while queue and cnt <= d:
for i in range(len(queue)):
x = queue.pop(0)
res.add(x)
for child in tree_gp[x]:
if child not in seen:
queue.append(child),seen.add(child)
cnt += 1
return res
res_set = { _ for _ in range(1,n+1)}
for i in range(m):
print(f_node[i])
print(bfs(tree_gp,f_node[i],d))
res_set = res_set & bfs(tree_gp,f_node[i],d)
print(len(res_set))
事后想了想,也不难啊,放到图里面做bfs,当时这么就没出来呢
查看原帖
点赞 1
相关推荐
点赞 评论 收藏
分享
01-29 19:56
西安电子科技大学 自然语言处理 点赞 评论 收藏
分享
牛客热帖
更多
- 1... 简历上的AI项目,面试官到底想看什么?2.0W
- 2... 字节java后端开发实习凉经7285
- 3... 我做过的,被面试官夸爆的那些Ai项目(二)5204
- 4... AI面试相关之RAG与Doris(JAVA)4636
- 5... 面试官视角聊聊:小龙虾OpenClaw如何0基础上手?3541
- 6... 航旅纵横JAVA 春招3493
- 7... AI产品实习生面试要达到什么水平?3400
- 8... 春招冲刺季|求职交流群正式启动!发帖赚现金,抱团拿offer!2931
- 9... 京东一面面经2897
- 10... 3.4 字节后端开发转正实习二面2080
正在热议
更多
# 交出你的校招焚诀 #
10076次浏览 171人参与
# 27届求职交流 #
2219次浏览 69人参与
# 神州信息求职进展汇总 #
3539次浏览 68人参与
# 实习生至暗时刻 #
17726次浏览 332人参与
# 26届求职交流 #
2136次浏览 52人参与
# 面试___岗的必刷题单 #
11803次浏览 208人参与
# 经纬恒润求职进展汇总 #
153235次浏览 1080人参与
# 哪些公司开暑期实习了? #
16690次浏览 137人参与
# 三月的小目标 #
10393次浏览 187人参与
# 你经历过哪些AI幻觉? #
4919次浏览 116人参与
# AI面试问题分享 #
12804次浏览 261人参与
# 春招开局,你有保底offer吗? #
24181次浏览 198人参与
# 公司情报交流地 #
144490次浏览 1273人参与
# 找AI工作应该卷什么? #
3850次浏览 68人参与
# 实习生的生存小技巧 #
6693次浏览 107人参与
# 米哈游求职进展汇总 #
583679次浏览 2997人参与
# 24届的你们现状如何了? #
112495次浏览 523人参与
# 实习想申请秋招offer,能不能argue薪资 #
224693次浏览 1193人参与
# 你的秋招第一面感觉怎么样 #
140510次浏览 805人参与
# 字节开奖 #
130580次浏览 602人参与
# 小厂一定不能去吗? #
23183次浏览 169人参与
