python - 计算结构有多深的最简单方法？

Question

假设我有一个像这样的 AST 结构(列表列表):

 [+, [*, a,b],[*,c,d] ]
 [+, [*, a,b],[*,c,[ +, d, e] ] ]

计算结构深度(即有多少层)的最简单和/或最快方法是什么？

作为第二个选项，您也可以将其作为字符串使用，而不是列表列表 (LoL)。例如:

"[+, [*, a,b],[*,c,d] ]"

我可以同时使用两者。

Answer 1

您可以保留方括号的数量。 您可以找到更详细的解释here 。 以下是代码的改编:

string = str(tree)

currentDepth = 0
maxDepth = 0
for c in string:
    if c == '[':
        currentDepth += 1
    elif c == ']':
        currentDepth -= 1

    maxDepth = max(maxDepth, currentDepth)

同样的警告，如果您的数据可能包含 '[' 或 ']'，这将会中断。 在这种情况下，您需要为这些方括号定义转义方法。