[LOJ6198] 謝特

之乎者助得甚？

給定字符串\(s\)和序列\(w\)，試求
\[ \max_{1\le i<j\le n} lcp(i,j)+(w_i\veebar w_j) \]
彷佛這樣的東西都能很好的用SA(height)+啓發式合併來完成？

能夠聯繫這題思考。

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;

int n,w[N],sa[N],ht[N],rc[N];
char s[N];

void buildSa() {
    static int c[N];
    int i,p,k,m=128,*x=ht,*y=rc;
    for(i=0; i<=m; ++i) c[i]=0;
    for(i=1; i<=n; ++i) c[x[i]=s[i]]++;
    for(i=1; i<=m; ++i) c[i]+=c[i-1];
    for(i=n; i>=1; --i) sa[c[x[i]]--]=i;
    for(k=1; k<n; k<<=1) {
        for(i=n-k+1,p=0; i<=n; ++i) y[++p]=i;
        for(i=1; i<=n; ++i) if(sa[i]>k) y[++p]=sa[i]-k;
        for(i=1; i<=m; ++i) c[i]=0;
        for(i=1; i<=n; ++i) c[x[y[i]]]++;
        for(i=1; i<=m; ++i) c[i]+=c[i-1];
        for(i=n; i>=1; --i) sa[c[x[y[i]]]--]=y[i];
        swap(x,y), x[sa[1]]=p=1;
        for(i=2; i<=n; ++i) x[sa[i]]=
            y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p;
        if((m=p)>=n) break;
    }
    for(i=1; i<=n; ++i) rc[sa[i]]=i;
    for(i=1,k=0; i<=n; ++i) {
        p=sa[rc[i]-1]; if(k) k--; 
        while(s[p+k]==s[i+k]) k++;
        ht[rc[i]]=k;
    }
}

int tot,top,stk[N*18];
int ll[N],rr[N],bl[N],siz[N],ch[N*18][2];
pair<int,int> h[N];

int newNode() {
    return top?stk[top--]:++tot;
}
void fuckNode(int x) {
    stk[++top]=x; 
    ch[x][0]=ch[x][1]=0;
}
void insert(int x,int w) {
    for(int i=16; ~i; --i) {
        if(!ch[x][(w>>i)&1]) ch[x][(w>>i)&1]=newNode();
        x=ch[x][(w>>i)&1];
    }
}
int ans,tmp;
void query(int x,int y,int d,int now) {
    if(d<0) {
        tmp=max(tmp,now);
        return;
    }
    for(int k=0; k<2; ++k) if(ch[x][k]) {
        if(ch[y][k^1]) query(ch[x][k],ch[y][k^1],d-1,now|(1<<d));
        else query(ch[x][k],ch[y][k],d-1,now);
    }
}
void merge(int x,int y,int d) {
    if(d<0) return;
    for(int k=0; k<2; ++k) if(ch[x][k]) {
        if(!ch[y][k]) ch[y][k]=newNode();
        merge(ch[x][k],ch[y][k],d-1);
        fuckNode(ch[x][k]);
    }
}

int main() {
//  freopen("C:/Users/HSY/Downloads/2.in","r",stdin);
    
    scanf("%d%s",&n,s+1);
    for(int i=1; i<=n; ++i) 
        scanf("%d",w+i),newNode();
    buildSa();
    for(int i=2; i<=n; ++i) h[i-1]=make_pair(-ht[i],i);
    for(int i=1; i<=n; ++i) {
        ll[i]=rr[i]=bl[i]=i; siz[i]=1;
        insert(i,w[sa[i]]);
    }
    sort(h+1,h+n);
    for(int i=1; i<n; ++i) {
        int len=-h[i].first;
        int x=bl[h[i].second];
        int y=bl[h[i].second-1];
        if(siz[x]>siz[y]) swap(x,y);
        tmp=0; 
        query(x,y,16,0);
        ans=max(ans,tmp+len);
        for(int j=ll[x]; j<=rr[x]; ++j) bl[j]=y;
        ll[y]=min(ll[x],ll[y]);
        rr[y]=max(rr[x],rr[y]);
        siz[y]+=siz[x];
        merge(x,y,16);
    }
    printf("%d\n",ans);
    return 0;
}