python - 后缀数组python的函数

Question

我想写一个输出后缀数组的函数。这是我目前所拥有的:

def suffixArray(s):
    sa = []
    for i in range(len(s)):
        suffix= sorted([s[i:]])
        sa = [len(s)-len(suffix[i:])
    return list(sa)

这会输出一个错误，因为我想我错过了一个额外的 if 语句，但我不确定如何去做。是的，我知道可能有更简单的方法来获取后缀数组，但我是 python 的初学者，我可以使用的函数很少。任何帮助表示赞赏。谢谢

这里还有一个我希望输入和输出的示例:输入 --> suffixArray('banana')输出--> [5, 3, 1, 0, 4, 2]

Answer 1

显然你想要每个后缀在按字典顺序排序后的索引

s = 'banana'

>>> [t[1] for t in sorted((s[i:],i) for i in range(len(s)))]
[5, 3, 1, 0, 4, 2]

或另一种方式:

>>> sorted(range(len(s)), key=lambda i: s[i:])
[5, 3, 1, 0, 4, 2]