Najdłuższe wspólne podciągi w czasie liniowym

45

To wyzwanie dotyczy pisania kodu w celu rozwiązania następującego problemu.

Biorąc pod uwagę dwa ciągi A i B, kod powinien wypisywać indeksy początkowe i końcowe podłańcucha A o następujących właściwościach.

  • Podciąg A powinien również pasować do niektórych podciągów B.
  • Nie powinno być już podłańcucha A, który spełnia pierwszą właściwość.

Na przykład:

A = xxxappleyyyyyyy

B = zapplezzz

Podłańcuch applez indeksami 4 8(indeksowanie od 1) byłby prawidłowym wyjściem.

Funkcjonalność

Możesz założyć, że dane wejściowe będą w standardzie w lub w pliku w katalogu lokalnym, który jest twoim wyborem. Format pliku to po prostu dwa ciągi znaków, oddzielone nowym wierszem. Odpowiedzią powinien być pełny program, a nie tylko funkcja.

W końcu chciałbym przetestować twój kod na dwóch podciągach pobranych z ciągów w http://hgdownload.cse.ucsc.edu/goldenPath/hg38/chromosomes/ .

Wynik

To jest golf golfowy z niespodzianką. Twój kod musi być uruchamiany w O(n)czasie, gdzie njest całkowitą długością danych wejściowych.

Języki i biblioteki

Możesz używać dowolnego języka, który ma swobodnie dostępny kompilator / tłumacz / etc. dla systemu Linux. Powinieneś używać tylko standardowych bibliotek open source nieprzeznaczonych do rozwiązania tego zadania. W przypadku sporu policzę to jako dowolną bibliotekę, która jest standardem w Twoim języku lub biblioteką, którą możesz zainstalować na domyślnej maszynie ubuntu z domyślnego repozytorium.

Przydatna informacja

Istnieją co najmniej dwa sposoby rozwiązania tego problemu w czasie liniowym. Jednym z nich jest najpierw obliczenie drzewa sufiksów, a drugim - najpierw obliczenie tablicy sufiksów i tablicy LCP.

Społeczność
źródło
4
O(n) timeJesteś pewien, że to możliwe?
Savenkov Alexey
17
@Lembik Przykro nam, ale są to bardzo złożone algorytmy i nie jest fajnie czytać ponad 100 linii kodu.
FUZxxl,
4
Artykuł pod drugim linkiem, który podajesz w „Przydatnych informacjach”, mówi, że „zbudowanie [drzewa sufiksów] wymaga czasu O (N ^ 2)”
KSFT
3
@Lembik Należy po prostu zadać pytanie [najszybszy kod], gdzie wygrywa program z najlepszym najgorszym przypadkiem w notacji big-oh. Wtedy przynajmniej dostaniesz kilka odpowiedzi, a nawet jeśli ktoś może rozwiązać go w O (n), wygra.
mbomb007,
9
To musi być pytanie z najczęściej
usuwanymi

Odpowiedzi:

39

Python 2, 646 bajtów

G=range;w=raw_input;z=L,m,h=[0]*3
s=w();t=len(s);s+='!%s#'%w();u=len(s);I=z*u
def f(s,n):
 def r(o):
    b=[[]for _ in s];c=[]
    for x in B[:N]:b[s[x+o]]+=x,
    map(c.extend,b);B[:N]=c
 M=N=n--~n/3;t=n%3%2;B=G(n+t);del B[::3];r(2);u=m=p=r(1)>r(0);N-=n/3
 for x in B*1:v=s[x:x+3];m+=u<v;u=v;B[x/3+x%3/2*N]=m
 A=1/M*z or f(B+z,M)+z;B=[x*3for x in A if x<N];J=I[r(0):n];C=G(n)
 for k in C:b=A[t]/N;a=i,j=A[t]%N*3-~b,B[p];q=p<N<(s[i:i-~b],J[i/3+b+N-b*N])>(s[j+t/M:j-~b],J[j/3+b*N]);C[k]=x=a[q];I[x]=k;p+=q;t+=1-q
 return C
S=f(map(ord,s)+z*40,u)
for i in G(u):
 h-=h>0;j=S[I[i]-1]
 while s[i+h]==s[j+h]:h+=1
 if(i<t)==(t<j)<=h>m:m=h;L=min(i,j)
print-~L,L+m

Wykorzystuje to algorytm skosu opisany w „Prostej konstrukcji tablicy sufiksu pracy liniowej” Kärkkäinena i Sandersa. Implementacja C ++ zawarta w tym dokumencie wydaje się już trochę „golfowa”, ale wciąż jest dużo miejsca na jej skrócenie. Na przykład możemy powtarzać się, dopóki nie dojdziemy do tablicy o długości jeden, zamiast zwarcia jak w dokumencie, bez naruszenia O(n)wymogu.

W części LCP śledziłem „Obliczenia najdłuższego wspólnego z prefiksem w czasie liniowym w tablicach sufiksów i jego zastosowaniach” autorstwa Kusai i in.

Program generuje wyjście, 1 0jeśli najdłuższy wspólny podciąg jest pusty.

Oto trochę kodu programistycznego, który zawiera wcześniejszą wersję programu, który jest nieco bliżej implementacji C ++, kilka wolniejszych podejść do porównania oraz prosty generator przypadków testowych:

from random import *

def brute(a,b):
    L=R=m=0

    for i in range(len(a)):
        for j in range(i+m+1,len(a)+1):
            if a[i:j] in b:
                m=j-i
                L,R=i,j

    return L+1,R

def suffix_array_slow(s):
    S=[]
    for i in range(len(s)):
        S+=[(s[i:],i)]
    S.sort()
    return [x[1] for x in S]

def slow1(a,b):
    # slow suffix array, slow lcp

    s=a+'!'+b
    S=suffix_array_slow(s)

    L=R=m=0

    for i in range(1,len(S)):
        x=S[i-1]
        y=S[i]
        p=s[x:]+'#'
        q=s[y:]+'$'
        h=0
        while p[h]==q[h]:
            h+=1
        if h>m and len(a)==sorted([x,y,len(a)])[1]:
            m=h
            L=min(x,y)
            R=L+h

    return L+1,R

def verify(a,b,L,R):
    if L<1 or R>len(a) or a[L-1:R] not in b:
        return 0
    LL,RR=brute(a,b)
    return R-L==RR-LL

def rand_string():
    if randint(0,1):
        n=randint(0,8)
    else:
        n=randint(0,24)
    a='zyxwvutsrq'[:randint(1,10)]
    s=''
    for _ in range(n):
        s+=choice(a)
    return s

def stress_test(f):
    numtrials=2000
    for trial in range(numtrials):
        a=rand_string()
        b=rand_string()
        L,R=f(a,b)
        if not verify(a,b,L,R):
            LL,RR=brute(a,b)
            print 'failed on',(a,b)
            print 'expected:',LL,RR
            print 'actual:',L,R
            return
    print 'ok'

def slow2(a,b):
    # slow suffix array, linear lcp

    s=a+'!'+b+'#'
    S=suffix_array_slow(s)

    I=S*1
    for i in range(len(S)):
        I[S[i]]=i

    L=R=m=h=0

    for i in range(len(S)):
        if I[i]:
            j=S[I[i]-1]
            while s[i+h]==s[j+h]:
                h+=1
            if h>m and len(a)==sorted([i,j,len(a)])[1]:
                m=h
                L=min(i,j)
                R=L+h
            h-=h>0

    return L+1,R

def suffix_array(s,K):
    # skew algorithm

    n=len(s)
    s+=[0]*3
    n0=(n+2)/3
    n1=(n+1)/3
    n2=n/3
    n02=n0+n2
    adj=n0-n1

    def radix_pass(a,o,n=n02):
        c=[0]*(K+3)
        for x in a[:n]:
            c[s[x+o]+1]+=1
        for i in range(K+3):
            c[i]+=c[i-1]
        for x in a[:n]:
            j=s[x+o]
            a[c[j]]=x
            c[j]+=1

    A=[x for x in range(n+adj) if x%3]+[0]*3

    radix_pass(A,2)
    radix_pass(A,1)
    radix_pass(A,0)

    B=[0]*n02
    t=m=0

    for x in A[:n02]:
        u=s[x:x+3]
        m+=t<u
        t=u
        B[x/3+x%3/2*n0]=m

    A[:n02]=1/n02*[0]or suffix_array(B,m)
    I=A*1
    for i in range(n02):
        I[A[i]]=i+1

    B=[3*x for x in A if x<n0]
    radix_pass(B,0,n0)

    R=[]

    p=0
    t=adj
    while t<n02:
        x=A[t]
        b=x>=n0
        i=(x-b*n0)*3-~b
        j=B[p]
        if p==n0 or ((s[i:i+2],I[A[t]-n0+1])<(s[j:j+2],I[j/3+n0]) if b else (s[i],I[A[t]+n0])<(s[j],I[j/3])):R+=i,;t+=1
        else:R+=j,;p+=1

    return R+B[p:n0]

def solve(a,b):
    # linear

    s=a+'!'+b+'#'
    S=suffix_array(map(ord,s),128)

    I=S*1
    for i in range(len(S)):
        I[S[i]]=i

    L=R=m=h=0

    for i in range(len(S)):
        if I[i]:
            j=S[I[i]-1]
            while s[i+h]==s[j+h]:
                h+=1
            if h>m and len(a)==sorted([i,j,len(a)])[1]:
                m=h
                L=min(i,j)
                R=L+h
            h-=h>0

    return L+1,R

stress_test(solve)
Mitch Schwartz
źródło
1
Popraw mnie, jeśli się mylę, ale czy to nie jest tak naprawdę 739 bajtów? Skopiowałem do mothereff.in/byte-counter i usunąłem 2 spacje z linii 6-9, ale nie jestem pewien, czy to prawda.
Patrick Roberts,
2
@PatrickRoberts To są zakładki.
Mitch Schwartz
2
Niezła odpowiedź! Może warto zajrzeć na GSACA, nowatorską liniową metodę SACA z 2016 roku. Implementacja referencyjna zawiera 246 linii pełnych komentarzy (170 bez komentarzy) i wydaje się bardzo grywalna. Znajdziesz go na github.
Christoph
1
@MitchSchwartz Obecnie próbuję pozostać na noPMO, więc nie mogę teraz silnie odczuwać emocji (prawdopodobnie z powodu niezrównoważonych chemikaliów w mózgu). W momencie szybkiego czytania kodu mój silnik gry w golfa zauważył to i nie pamiętam żadnych konkretnych emocji. Czy myślałeś o tym samym lub dlaczego pytanie? :) Teraz jestem ciekawa.
Yytsi
1
@TuukkaX To ciekawa odpowiedź, której się nie spodziewałem. Cóż, nie jestem pewien, czy powinienem frazować to w jakiś szczególny sposób, ale fakt, że twój oryginalny komentarz nie był właściwie poprawny, odegrał pewną rolę w tym, dlaczego zdecydowałem się zapytać. :)
Mitch Schwartz