找到最长的相邻重复且不重叠的子字符串

6
(这个问题与音乐无关,只是以音乐作为用例举例说明。)
在音乐中,常见的短语结构方式是将一系列音符组成一个循环部分,并重复一次或多次作为中间部分。因此,短语包括引言、循环部分和结尾。以下是一个示例:
[ E E E F G A F F G A F F G A F C D ]

我们可以“看到”引言部分是 [ E E E ],重复部分是 [ F G A F ],结束部分是 [ C D ]。因此,拆分列表的方法是:
[ [ E E E ] 3 [ F G A F ] [ C D ] ]

其中第一个项目是介绍,第二个数字表示重复部分的重复次数,第三个部分是结尾。

我需要一个算法来执行这样的拆分。

但是有一个注意点,即可能有多种方式来拆分列表。例如,上面的列表可以被拆分为:

[ [ E E E F G A ] 2 [ F F G A ] [ F C D ] ]

但这是一种更糟糕的分割方式,因为引言和结束语更长。因此,算法的标准是找到最大化循环部分长度并同时最小化引言和结束语总长度的分割点。这意味着正确的分割点应该是

[ A C C C C C C C C C A ]

is

[ [ A ] 9 [ C ] [ A ] ]

由于引言和结尾的总长度为2,循环部分的长度为9。

此外,虽然引言和结尾可以为空,但只允许“真正”的重复。因此,以下拆分是不允许的:

[ [ ] 1 [ E E E F G A F F G A F F G A F C D ] [ ] ]

将其视为寻找序列的最佳“压缩”方式。请注意,某些序列中可能没有任何重复项:

[ A B C D ]

对于这些退化情况,允许任何合理的结果。

这是我的算法实现:

def find_longest_repeating_non_overlapping_subseq(seq):
    candidates = []
    for i in range(len(seq)):
        candidate_max = len(seq[i + 1:]) // 2
        for j in range(1, candidate_max + 1):
            candidate, remaining = seq[i:i + j], seq[i + j:]
            n_reps = 1
            len_candidate = len(candidate)
            while remaining[:len_candidate] == candidate:
                n_reps += 1
                remaining = remaining[len_candidate:]
            if n_reps > 1:
                candidates.append((seq[:i], n_reps,
                                   candidate, remaining))
    if not candidates:
        return (type(seq)(), 1, seq, type(seq)())

    def score_candidate(candidate):
        intro, reps, loop, outro = candidate
        return reps - len(intro) - len(outro)
    return sorted(candidates, key = score_candidate)[-1]

我不确定这是否正确,但它通过了我所描述的简单测试。问题在于它太慢了。我已经研究了后缀树,但它们似乎不适合我的用例,因为我需要的子字符串应该是不重叠且相邻的。

3个回答

6

这是一种明显具有二次时间复杂度的方法,但由于它不建立除长度为1的子字符串之外的任何子字符串对象,所以具有相对较低的常数因子。结果是一个2元组,

bestlen, list_of_results

其中bestlen是重复相邻块的最长子字符串长度,每个结果都是一个3元组,

start_index, width, numreps

这意味着被重复的子字符串是

the_string[start_index : start_index + width]

这些相邻元素中有numreps个。这总是如此。

bestlen == width * numreps

问题描述存在歧义。例如,考虑以下输出:
>>> crunch2("aaaaaabababa")
(6, [(0, 1, 6), (0, 2, 3), (5, 2, 3), (6, 2, 3), (0, 3, 2)])

因此,它找到了5种将“最长”的连续部分视为长度为6的方法:

  • 将初始的“a”重复6次。
  • 将初始的“aa”重复3次。
  • 将最左边的“ab”重复3次。
  • 将最左边的“ba”重复3次。
  • 将初始的“aaa”重复2次。

它不返回介绍或结尾,因为这些可以从它返回的内容中轻易推断出来:

  • 引言是the_string[: start_index]
  • 结尾是the_string[start_index + bestlen :]

如果没有重复相邻块,则返回

(0, [])

其他例子(来自您的帖子):

>>> crunch2("EEEFGAFFGAFFGAFCD")
(12, [(3, 4, 3)])
>>> crunch2("ACCCCCCCCCA")
(9, [(1, 1, 9), (1, 3, 3)])
>>> crunch2("ABCD")
(0, [])

如何实现的关键:假设您有相邻重复的块,每个块的宽度为 W。然后考虑当您将原始字符串与向左移动 W 的字符串进行比较时会发生什么:
... block1 block2 ... blockN-1 blockN ...
... block2 block3 ... blockN      ... ...

如果你在同一位置获得了(N-1)*W个连续相等的字符,那么问题就解决了。但是这个方法也可以反过来使用:如果你向左移动W并找到(N-1)*W个连续相等的字符,那么你就可以得出结论:

block1 == block2
block2 == block3
...
blockN-1 == blockN

所以所有的 N 块必须是 block1 的重复。

因此,代码重复将原始字符串向左移动一个字符(的副本),然后从左到右地遍历两个字符并标识相等字符的最长连续区间。这只需要每次比较一对字符。为了使“向左移”高效(常数时间),字符串的副本存储在 collections.deque 中。

注:在许多情况下,update() 执行了太多无用的工作,现已被替换。

def crunch2(s):
    from collections import deque

    # There are zcount equal characters starting
    # at index starti.
    def update(starti, zcount):
        nonlocal bestlen
        while zcount >= width:
            numreps = 1 + zcount // width
            count = width * numreps
            if count >= bestlen:
                if count > bestlen:
                    results.clear()
                results.append((starti, width, numreps))
                bestlen = count
            else:
                break
            zcount -= 1
            starti += 1

    bestlen, results = 0, []
    t = deque(s)
    for width in range(1, len(s) // 2 + 1):
        t.popleft()
        zcount = 0
        for i, (a, b) in enumerate(zip(s, t)):
            if a == b:
                if not zcount: # new run starts here
                    starti = i
                zcount += 1
            # else a != b, so equal run (if any) ended
            elif zcount:
                update(starti, zcount)
                zcount = 0
        if zcount:
            update(starti, zcount)
    return bestlen, results

使用正则表达式

[由于大小限制已删除]

使用后缀数组

这是目前我找到的最快的方法,尽管仍然可能会导致二次时间复杂度。

请注意,找到重叠的字符串并不重要。如上面的crunch2()程序中解释的那样(在一些细节上解释),

  • 给定字符串s,其长度为n = len(s)
  • 给定整数ij,其中 0 <= i < j < n

那么如果w = j-i,并且cs[i:]s[j:]之间共同前导字符的数量,则块s[i:j](长度为w)从s[i]开始重复,共1 + c // w 次。

下面的程序直接遵循这个规则来查找所有重复相邻块,并记住总长度最大的那些块。返回与crunch2()相同的结果,但有时顺序不同。

后缀数组简化了搜索,但几乎没有消除它。后缀数组直接找到具有最大c<i, j>, 但只限制搜索以最大化w * (1 + c // w)。最坏情况是形如letter * number的字符串,例如"a" * 10000

我没有提供下面的sa模块的代码。它很啰嗦,任何后缀数组的实现都会计算相同的东西。suffix_array()的输出如下:

  • sa是后缀数组,是range(n)的唯一排列,使得对于所有irange(1, n)中,s[sa[i-1]:] < s[sa[i]:]

  • rank在这里没有使用。

  • 对于range(1, n)中的ilcp[i]给出从sa[i-1]sa[i]开始的后缀之间的最长公共前缀的长度。

为什么Suffix Array胜出呢? 部分原因是它从不需要搜索以相同字母开头的后缀(通过构造,后缀数组使它们相邻),而检查重复块和是否为新的最佳值,无论块有多大或重复了多少次,都只需花费小的常数时间。如上所述,这只需在 C 和 w 上进行微不足道的算术计算即可。
免责声明: 对于我来说,后缀数组/树就像连续分数一样: 我能在必要时使用它们,并对结果感到惊奇,但它们让我头疼。敏感,敏感,敏感。
def crunch4(s):
    from sa import suffix_array
    sa, rank, lcp = suffix_array(s)
    bestlen, results = 0, []
    n = len(s)
    for sai in range(n-1):
        i = sa[sai]
        c = n
        for saj in range(sai + 1, n):
            c = min(c, lcp[saj])
            if not c:
                break
            j = sa[saj]
            w = abs(i - j)
            if c < w:
                continue
            numreps = 1 + c // w
            assert numreps > 1
            total = w * numreps
            if total >= bestlen:
                if total > bestlen:
                    results.clear()
                    bestlen = total
                results.append((min(i, j), w, numreps))
    return bestlen, results

一些时间记录

我将一个包含英语单词的小文件读入字符串 xs 中,每行一个单词:

>>> len(xs)
209755
>>> xs.count('\n')
25481

大约有25,000个单词,占用210,000个字节。这些算法的时间复杂度是二次方级别,所以我并不指望它运行得很快,但是经过数小时后,crunch2()仍在运行,而当我让它过夜时,它仍在运行。

这导致我意识到它的update()函数可能会做大量无用功,使得整个算法的时间复杂度变成立方级别。因此,我修复了这个问题。然后:

>>> crunch2(xs)
(44, [(63750, 22, 2)])
>>> xs[63750 : 63750+50]
'\nelectroencephalograph\nelectroencephalography\nelec'

这大约花费了38分钟,与我的预期相符。

正则表达式版本的crunch3()只需要不到十分之一秒的时间!

>>> crunch3(xs)
(8, [(19308, 4, 2), (47240, 4, 2)])
>>> xs[19308 : 19308+10]
'beriberi\nB'
>>> xs[47240 : 47240+10]
'couscous\nc'

如前所述,正则表达式版本可能无法找到最佳答案,但这里有另外一些因素: 默认情况下,“.”不匹配换行符,所以代码实际上进行了许多微小的搜索。文件中的每个约 25K 换行符都有效地结束了本地搜索范围。用re.DOTALL标志编译正则表达式代替(因此换行符不被特殊处理):

>>> crunch3(xs) # with DOTALL
(44, [(63750, 22, 2)])

大约14分钟后。

最后,

>>> crunch4(xs)
(44, [(63750, 22, 2)])

在不到9分钟的时间内,构建后缀数组所需的时间非常微小(不到一秒钟)。这实际上相当令人印象深刻,因为即使是 brute force 正则表达式版本几乎完全“以 C 速度”运行,但速度仍比较慢。
但这是相对的。从绝对意义上讲,所有这些都还是非常慢的 :-(
注意:下一节中的版本将把时间缩短到不到5秒(!)。

巨大的加速

这个方法采用了完全不同的方法。 对于上面的大型字典示例,它可以在不到5秒的时间内得出正确答案。
我为此感到相当自豪;-) 这是出乎意料的,我还没有见过这种方法。 它不执行任何字符串搜索,只对索引集进行整数算术运算。
对于类似于 letter * largish_integer 的输入格式,它仍然非常缓慢。目前的做法是只要当前考虑长度的子串中至少存在两个(不一定是相邻的,甚至可能重叠的!)副本,就会增加 1。例如,在...中。
'x' * 1000000

它将尝试从1到999999的所有子字符串大小。

然而,似乎可以通过重复加倍当前大小(而不仅仅是添加1),保存每个步骤中的类别,并进行混合二分搜索来定位存在重复的最大子字符串大小,从而显著提高效率。

我会留下这个无疑枯燥无味的练习给读者完成。我的工作已经完成了;-)

def crunch5(text):
    from collections import namedtuple, defaultdict

    # For all integers i and j in IxSet x.s,
    # text[i : i + x.w] == text[j : j + x.w].
    # That is, it's the set of all indices at which a specific
    # substring of length x.w is found.
    # In general, we only care about repeated substrings here,
    # so weed out those that would otherwise have len(x.s) == 1.
    IxSet = namedtuple("IxSet", "s w")

    bestlen, results = 0, []

    # Compute sets of indices for repeated (not necessarily
    # adjacent!) substrings of length xs[0].w + ys[0].w, by looking
    # at the cross product of the index sets in xs and ys.
    def combine(xs, ys):
        xw, yw = xs[0].w, ys[0].w
        neww = xw + yw
        result = []
        for y in ys:
            shifted = set(i - xw for i in y.s if i >= xw)
            for x in xs:
                ok = shifted & x.s
                if len(ok) > 1:
                    result.append(IxSet(ok, neww))
        return result

    # Check an index set for _adjacent_ repeated substrings.
    def check(s):
        nonlocal bestlen
        x, w = s.s.copy(), s.w
        while x:
            current = start = x.pop()
            count = 1
            while current + w in x:
                count += 1
                current += w
                x.remove(current)
            while start - w in x:
                count += 1
                start -= w
                x.remove(start)
            if count > 1:
                total = count * w
                if total >= bestlen:
                    if total > bestlen:
                        results.clear()
                        bestlen = total
                    results.append((start, w, count))

    ch2ixs = defaultdict(set)
    for i, ch in enumerate(text):
        ch2ixs[ch].add(i)
    size1 = [IxSet(s, 1)
             for s in ch2ixs.values()
             if len(s) > 1]
    del ch2ixs
    for x in size1:
        check(x)

    current_class = size1
    # Repeatedly increase size by 1 until current_class becomes
    # empty. At that point, there are no repeated substrings at all
    # (adjacent or not) of the then-current size (or larger).
    while current_class:
        current_class = combine(current_class, size1)
        for x in current_class:
            check(x)
    
    return bestlen, results

更快的算法

crunch6() 程序在我的计算机上能够将一个大型字典实例处理到不到 2 秒。它结合了来自 crunch4() (后缀和 lcp 数组)和 crunch5()(在一组索引中查找给定步幅的所有算术级数)的思想。

crunch5() 相似,代码循环次数等于最长重复子串的长度加1(可以是重叠或者不重叠)。如果没有长度为n 的重复,则没有长度大于n 的重复。这样做会使寻找重复字符串更容易,因为这是一个可利用的限制条件。但是,当将“胜利”约束在相邻重复的情况下,这种方法就行不通了。例如,在“abcabc”中,没有长度为1的偶数相邻重复,但有一个长度为3的相邻重复。这样做貌似使得任何直接二分搜索都是徒劳无功的(相邻大小为n的重复的存在或不存在并不能说明是否存在其他大小的相邻重复)。

对于形如'x' * n 的输入,效果仍然不佳。其中包含了所有长度从1到n-1的重复字符串。

注意:我给出的所有程序都能够生成所有最大长度相邻块的分解方式。例如,在一个由9个“x”组成的字符串中,它表示它可以通过重复“x” 9 次或者重复“xxx” 3 次来得到。因此,令人惊奇的是,它们都可以用作分解算法;-)

def crunch6(text):
    from sa import suffix_array
    sa, rank, lcp = suffix_array(text)
    bestlen, results = 0, []
    n = len(text)

    # Generate maximal sets of indices s such that for all i and j
    # in s the suffixes starting at s[i] and s[j] start with a
    # common prefix of at least len minc.
    def genixs(minc, sa=sa, lcp=lcp, n=n):
        i = 1
        while i < n:
            c = lcp[i]
            if c < minc:
                i += 1
                continue
            ixs = {sa[i-1], sa[i]}
            i += 1
            while i < n:
                c = min(c, lcp[i])
                if c < minc:
                    yield ixs
                    i += 1
                    break
                else:
                    ixs.add(sa[i])
                    i += 1
            else: # ran off the end of lcp
                yield ixs

    # Check an index set for _adjacent_ repeated substrings
    # w apart.  CAUTION: this empties s.
    def check(s, w):
        nonlocal bestlen
        while s:
            current = start = s.pop()
            count = 1
            while current + w in s:
                count += 1
                current += w
                s.remove(current)
            while start - w in s:
                count += 1
                start -= w
                s.remove(start)
            if count > 1:
                total = count * w
                if total >= bestlen:
                    if total > bestlen:
                        results.clear()
                        bestlen = total
                    results.append((start, w, count))

    c = 0
    found = True
    while found:
        c += 1
        found = False
        for s in genixs(c):
            found = True
            check(s, c)
    return bestlen, results

总是快速且发布,但有时错误

在生物信息学中,这被研究为“串联重复”、“串联阵列”和“简单序列重复”(SSR)。您可以搜索这些术语以查找相当多的学术论文,其中一些声称具有最坏情况下的线性时间算法。

但是它们似乎分为两个阵营:

  1. 类似于上面的crunch4(),但没有它的内部循环的那种线性时间算法实际上是错误的 :-(
  2. 算法过于复杂,即使尝试将它们转换为可运行代码都需要付出努力 :-(

在第一个阵营中,有几篇论文可以归结为crunch4a(),但是没有其内部循环。下面是一个例子:crunch4a()。这里有一个例子:

“SA-SSR:基于后缀数组的大型遗传序列中详尽且高效的SSR发现算法。”

Pickett等人

https://www.ncbi.nlm.nih.gov/pmc/articles/PMC5013907/

crunch4a()总是快速的,但有时会出错。实际上,它为这里出现的每个示例找到了至少一个最大重复段,将较大的字典示例解决在几秒钟内,并且对于形式为'x' * 1000000的字符串没有问题。大部分时间都用于构建后缀和LCP数组。但是它可能失败:

>>> x = "bcdabcdbcd"
>>> crunch4(x)  # finds repeated bcd at end
(6, [(4, 3, 2)])
>>> crunch4a(x) # finds nothing
(0, [])

问题在于不能保证“相关”的后缀在后缀数组中是相邻的。以“b”开头的后缀排序如下:
bcd
bcdabcdbcd
bcdbcd

通过这种方法找到尾部重复块需要将第一个块与第三个块进行比较。这就是为什么crunch4()有一个内部循环,尝试所有以相同字母开头的对。相关的对可以在后缀数组中的任意数量的其他后缀中分开。但这也使算法成为二次时间。
# only look at adjacent entries - fast, but sometimes wrong
def crunch4a(s):
    from sa import suffix_array
    sa, rank, lcp = suffix_array(s)
    bestlen, results = 0, []
    n = len(s)
    for sai in range(1, n):
        i, j = sa[sai - 1], sa[sai]
        c = lcp[sai]
        w = abs(i - j)
        if c >= w:
            numreps = 1 + c // w
            total = w * numreps
            if total >= bestlen:
                if total > bestlen:
                    results.clear()
                    bestlen = total
                results.append((min(i, j), w, numreps))
    return bestlen, results

O(n log n)

这篇论文看起来对我来说是正确的,尽管我没有编写它:

“Simple and Flexible Detection of Contiguous Repeats Using a Suffix Tree”

Jens Stoye, Dan Gusfield

https://csiflabs.cs.ucdavis.edu/~gusfield/tcs.pdf

要实现亚二次算法需要做出一些妥协。例如,"x" * n 具有形如 "x"*2"x"*3 等的 n-1 个子字符串,因此仅这些字符串就有 O(n**2) 种。 因此,任何找到所有这些字符串的算法都必须至少是二次时间复杂度。

详细信息请参阅论文;-) 您要寻找的一个概念是“原语”:“S*n” 的重复只能表示为不能再用较短的字符串重复表达的S。例如,"x" * 10 是 primitive,但 "xx" * 5 不是。

通向 O(n log n) 的一步

crunch9() 是我在评论中提到的“暴力”算法的实现,来自于:

“The enhanced suffix array and its applications to genome analysis”

Ibrahim等人

http://citeseerx.ist.psu.edu/viewdoc/download?doi=10.1.1.93.2217&rep=rep1&type=pdf

那里的实现草图只能找出“分支串联”重复项,而我在这里添加了代码来推断任意数量的重复项,并包括非分支重复项。虽然它最坏情况下仍是 O(n**2),但对于您在评论中提到的 seq 字符串来说,它比这里的其他任何程序都快得多。目前为止,它与大多数其他程序产生了(除了顺序外)相同的详尽报告。

论文继续努力将最坏情况削减为 O(n log n),但这会使其变慢。因此,它更加努力地斗争。我承认我失去了兴趣;-)

# Generate lcp intervals from the lcp array.
def genlcpi(lcp):
    lcp.append(0)
    stack = [(0, 0)]
    for i in range(1, len(lcp)):
        c = lcp[i]
        lb = i - 1
        while c < stack[-1][0]:
            i_c, lb = stack.pop()
            interval = i_c, lb, i - 1
            yield interval
        if c > stack[-1][0]:
            stack.append((c, lb))
    lcp.pop()

def crunch9(text):
    from sa import suffix_array

    sa, rank, lcp = suffix_array(text)
    bestlen, results = 0, []
    n = len(text)

    # generate branching tandem repeats
    def gen_btr(text=text, n=n, sa=sa):
        for c, lb, rb in genlcpi(lcp):
            i = sa[lb]
            basic = text[i : i + c]
            # Binary searches to find subrange beginning with
            # basic+basic. A more gonzo implementation would do this
            # character by character, never materialzing the common
            # prefix in `basic`.
            rb += 1
            hi = rb
            while lb < hi:  # like bisect.bisect_left
                mid = (lb + hi) // 2
                i = sa[mid] + c
                if text[i : i + c] < basic:
                    lb = mid + 1
                else:
                    hi = mid
            lo = lb
            while lo < rb:  # like bisect.bisect_right
                mid = (lo + rb) // 2
                i = sa[mid] + c
                if basic < text[i : i + c]:
                    rb = mid
                else:
                    lo = mid + 1
            lead = basic[0]
            for sai in range(lb, rb):
                i = sa[sai]
                j = i + 2*c
                assert j <= n
                if j < n and text[j] == lead:
                    continue # it's non-branching
                yield (i, c, 2)

    for start, c, _ in gen_btr():
        # extend left
        numreps = 2
        for i in range(start - c, -1, -c):
            if all(text[i+k] == text[start+k] for k in range(c)):
                start = i
                numreps += 1
            else:
                break
        totallen = c * numreps
        if totallen < bestlen:
            continue
        if totallen > bestlen:
            bestlen = totallen
            results.clear()
        results.append((start, c, numreps))
        # add non-branches
        while start:
            if text[start - 1] == text[start + c - 1]:
                start -= 1
                results.append((start, c, numreps))
            else:
                break
    return bestlen, results

获得奖励积分 ;-)

对于一些技术含义而言,crunch11() 的最坏情况复杂度为 O(n log n)。除了后缀和最长公共前缀数组之外,还需要 rank 数组以及 sa 的逆序数组。

assert all(rank[sa[i]] == sa[rank[i]] == i for i in range(len(sa)))

正如代码注释所述,该程序还依赖于Python 3以提高速度(range()的行为)。这很浅显易懂,但要重写它将非常繁琐。

描述该程序的论文存在一些错误,因此如果该代码与您所阅读的内容不完全相符,请不要惊慌。相反,请按照他们所说的来实现,否则就会失败。

尽管如此,这段代码变得越来越复杂,我无法保证其中没有漏洞。但是,在我尝试过的所有情况下,它都能正常工作。

形如'x' * 1000000的输入仍然不够快,但显然不再是二次时间。例如,一个重复出现一百万次同一字母的字符串可以在接近30秒内完成。大多数其他程序则永远无法结束;-)

编辑:将genlcpi()更改为使用半开放式的Python范围;对crunch11()进行了大多数外观上的修改;增加了“提前退出”功能,可以节省最差情况(例如'x' * 1000000)下约三分之一的时间。

# Generate lcp intervals from the lcp array.
def genlcpi(lcp):
    lcp.append(0)
    stack = [(0, 0)]
    for i in range(1, len(lcp)):
        c = lcp[i]
        lb = i - 1
        while c < stack[-1][0]:
            i_c, lb = stack.pop()
            yield (i_c, lb, i)
        if c > stack[-1][0]:
            stack.append((c, lb))
    lcp.pop()

def crunch11(text):
    from sa import suffix_array

    sa, rank, lcp = suffix_array(text)
    bestlen, results = 0, []
    n = len(text)

    # Generate branching tandem repeats.
    # (i, c, 2) is branching tandem iff
    #     i+c in interval with prefix text[i : i+c], and
    #     i+c not in subinterval with prefix text[i : i+c + 1]
    # Caution: this pragmatically relies on that, in Python 3,
    # `range()` returns a tiny object with O(1) membership testing.
    # In Python 2 it returns a list - ahould still work, but very
    # much slower.
    def gen_btr(text=text, n=n, sa=sa, rank=rank):
        from itertools import chain

        for c, lb, rb in genlcpi(lcp):
            origlb, origrb = lb, rb
            origrange = range(lb, rb)
            i = sa[lb]
            lead = text[i]
            # Binary searches to find subrange beginning with
            # text[i : i+c+1]. Note we take slices of length 1
            # rather than just index to avoid special-casing for
            # i >= n.
            # A more elaborate traversal of the lcp array could also
            # give us a list of child intervals, and then we'd just
            # need to pick the right one. But that would be even
            # more hairy code, and unclear to me it would actually
            # help the worst cases (yes, the interval can be large,
            # but so can a list of child intervals).
            hi = rb
            while lb < hi:  # like bisect.bisect_left
                mid = (lb + hi) // 2
                i = sa[mid] + c
                if text[i : i+1] < lead:
                    lb = mid + 1
                else:
                    hi = mid
            lo = lb
            while lo < rb:  # like bisect.bisect_right
                mid = (lo + rb) // 2
                i = sa[mid] + c
                if lead < text[i : i+1]:
                    rb = mid
                else:
                    lo = mid + 1
            subrange = range(lb, rb)
            if 2 * len(subrange) <= len(origrange):
                # Subrange is at most half the size.
                # Iterate over it to find candidates i, starting
                # with wa.  If i+c is also in origrange, but not
                # in subrange, good:  then i is of the form wwx.
                for sai in subrange:
                    i = sa[sai]
                    ic = i + c
                    if ic < n:
                        r = rank[ic]
                        if r in origrange and r not in subrange:
                            yield (i, c, 2, subrange)
            else:
                # Iterate over the parts outside subrange instead.
                # Candidates i are then the trailing wx in the
                # hoped-for wwx. We win if i-c is in subrange too
                # (or, for that matter, if it's in origrange).
                for sai in chain(range(origlb, lb),
                                 range(rb, origrb)):
                    ic = sa[sai] - c
                    if ic >= 0 and rank[ic] in subrange:
                        yield (ic, c, 2, subrange)

    for start, c, numreps, irange in gen_btr():
        # extend left
        crange = range(start - c, -1, -c)
        if (numreps + len(crange)) * c < bestlen:
            continue
        for i in crange:
            if rank[i] in irange:
                start = i
                numreps += 1
            else:
                break
        # check for best
        totallen = c * numreps
        if totallen < bestlen:
            continue
        if totallen > bestlen:
            bestlen = totallen
            results.clear()
        results.append((start, c, numreps))
        # add non-branches
        while start and text[start - 1] == text[start + c - 1]:
                start -= 1
                results.append((start, c, numreps))
    return bestlen, results

非常感谢您详细的回答!我需要一些时间来消化它。 :) 看起来我遇到了非常复杂的问题。正如您所说,我的规范是含糊不清的,因为我不想限制任何人的创造力。我的想法是递归地使用算法来构建字符串的树形表示。因此,对于 aaaaaa[ [] 6 a [] ][ [] 3 aa [] ] 更好,因为在递归之后,您将得到 [ [] 3 [ [] 2 a [] ] [] ] - Björn Lindqvist
在这种情况下,您可能会喜欢这篇论文 :-) 不幸的是,我没有找到免费版本(我也没有阅读过!)。 "查找字符串的最小重复表示的算法" https://link.springer.com/chapter/10.1007/978-3-642-16321-0_18 - Tim Peters
1
由于某些原因,我从您的算法中得到的性能不如预期。这里是一个要点,其中包含了在仅有4k个字符的字符串上运行crunch6所需的4秒时间。该要点中的字符串似乎触发了一些退化情况。 - Björn Lindqvist
顺便提一下,针对你的“seq”,我的原始“crunch2()”比“crunch6()”运行速度显著更快。在任何需要“crunch6()”运行的通道数量(max(lcp)+1)接近(或大于)len(text)/2 的情况下都是如此 - 每个“crunch2”通道要简单得多比“crunch6”通道。 - Tim Peters
请查看新的 crunch9() 编辑,它对于您的 seq(以及我的字典示例)非常快。 - Tim Peters
显示剩余5条评论

1
这是我对你所说的实现方式。它与你的相似,但会跳过已被检查为先前子字符串重复的子字符串。
from collections import namedtuple
SubSequence = namedtuple('SubSequence', ['start', 'length', 'reps'])

def longest_repeating_subseq(original: str):
    winner = SubSequence(start=0, length=0, reps=0)
    checked = set()
    subsequences = (  # Evaluates lazily during iteration
        SubSequence(start=start, length=length, reps=1)
        for start in range(len(original))
        for length in range(1, len(original) - start)
        if (start, length) not in checked)

    for s in subsequences:
        subseq = original[s.start : s.start + s.length]
        for reps, next_start in enumerate(
                range(s.start + s.length, len(original), s.length),
                start=1):
            if subseq != original[next_start : next_start + s.length]:
                break
            else:
                checked.add((next_start, s.length))

        s = s._replace(reps=reps)
        if s.reps > 1 and (
                (s.length * s.reps > winner.length * winner.reps)
                or (  # When total lengths are equal, prefer the shorter substring
                    s.length * s.reps == winner.length * winner.reps
                    and s.reps > winner.reps)):
            winner = s

    # Check for default case with no repetitions
    if winner.reps == 0:
        winner = SubSequence(start=0, length=len(original), reps=1)

    return (
        original[ : winner.start],
        winner.reps,
        original[winner.start : winner.start + winner.length],
        original[winner.start + winner.length * winner.reps : ])

def test(seq, *, expect):
    print(f'Testing longest_repeating_subseq for {seq}')
    result = longest_repeating_subseq(seq)
    print(f'Expected {expect}, got {result}')
    print(f'Test {"passed" if result == expect else "failed"}')
    print()

if __name__ == '__main__':
    test('EEEFGAFFGAFFGAFCD', expect=('EEE', 3, 'FGAF', 'CD'))
    test('ACCCCCCCCCA', expect=('A', 9, 'C', 'A'))
    test('ABCD', expect=('', 1, 'ABCD', ''))

对我来说,您的三个示例都通过了。这似乎是一种可能具有许多奇怪边缘情况的事情,但考虑到它是优化过的蛮力方法,更可能是更新规范而不是修复代码本身中的错误。


0

看起来你正在尝试使用LZ77压缩算法。你可以参考我提供的维基百科文章中的参考实现,检查你的代码。


1
我预计他们会遇到与后缀树相同的问题(有时用于实现LZ方案):LZ不关心“相邻”或“非重叠”。 - Tim Peters

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接