我有一个python对象,结构如下:
tree = {
"a": {
"aa": {
"aaa": {},
"aab": {},
"aac": {}
},
"ab": {
"aba": {},
"abb": {}
}
},
"b": {
"ba": {}
},
"c": {}
}
一组字符串列表,如:
arr_1 = ["a", "ab"]
arr_2 = ["b", "ab"]
arr_3 = ["a", "ab", "aba"]
每个列表都定义了树中键的路径,我想确定列表是否描述了树中的有效路径。
目前,我可以像这样逐案回答:
tree[arr_1[0]][arr_1[1]] # Returns correct object
tree[arr_2[0]][arr_2[1]] # Returns error
tree[arr_3[0]][arr_3[1]][arr_3[2]] # Returns correct object
虽然这不符合我的要求。我更喜欢一个函数,它会在树中搜索任何给定列表中的键。
以下功能几乎是我想要的,但它不处理不同长度的列表。
def is_valid(tree, arr):
obj = tree[arr[0]][arr[1]]
if len(obj) > 0:
return(obj)
else:
return("No obj")
目前此功能输出
is_valid(tree, arr_1) # Returns the correct result
is_valid(tree, arr_2) # Returns an error
is_valid(tree, arr_3) # Returns the same result as arr_1, which is incorrect
任何人都可以帮我扩展这个功能,以动态地对arr
参数的长度作出反应吗?
谢谢!
答案 0 :(得分:3)
我认为最简单的方法是使用递归。每个子树都是树,每个子路径都是路径,因此我们可以查看路径中的第一个元素是否有效,然后从那里继续。
def is_valid(tree, path):
# base case. Any path would be valid for any tree
if len(path) == 0:
return True
# the path must be invalid, no matter what comes before or after
if not path[0] in tree:
return False
# look at the rest
return is_valid(tree[path[0]], path[1:])
如果您希望子树描述路径,则可以执行此操作:
def is_valid(tree, path):
if len(path) == 0:
return tree # the only difference
if not path[0] in tree:
return False
return is_valid(tree[path[0]], path[1:])