bzoj1396: 识别子串

1396: 识别子串

Description

 

---

Solution

只出现一次的点只能是叶子。

假设某一个叶子表示的区间是[1-Max]

它的父亲的长度是M。 也就是[Max-M,Max]只出现过一次。

那么我们可以用这M+1长来更新[Max-M,Max]。----A

剩下[1,Max-M]的位置x，可以用[x,Max]这段区间的长度来更新。---B

于是按下标开两棵线段树，一棵维护权值，实现A，一棵维护位置，实现B

然后就好啦

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define maxn 200005
using namespace std;
int n,cnt=1,rt=1,la=1,w[maxn],tr[maxn*20][2],root[maxn],tot,ans;
int num[maxn];
char ch[maxn];
struct node{
    int Max,par,nex[26],id;    
}s[maxn];
vector<int>G[maxn];
void ins(int c,int id){
    int np=++cnt,p=la;la=np;
    s[np].Max=s[p].Max+1;s[np].id=id;
    for(;p&&!s[p].nex[c];p=s[p].par)s[p].nex[c]=np;
    if(!p)s[np].par=rt;
    else {
        int q=s[p].nex[c];
        if(s[q].Max==s[p].Max+1)s[np].par=q;
        else {
            int nq=++cnt;
            s[nq].Max=s[p].Max+1;s[nq].par=s[q].par;
            for(int i=0;i<26;i++)s[nq].nex[i]=s[q].nex[i];
            s[q].par=s[np].par=nq;
            for(;s[p].nex[c]==q;p=s[p].par)s[p].nex[c]=nq;
        }
    }
} 


int get(int v,int w){
    return (v&(1<<w))>0;
}
void add(int &R,int v){
    if(!R)R=++tot;
    int k=R;
    for(int i=20;i>=0;i--){
        int p=get(v,i);
        if(!tr[k][p])tr[k][p]=++cnt;
        k=tr[k][p];
    }
}
int merge(int x,int y){
    if(!x||!y)return x+y;
    tr[x][0]=merge(tr[x][0],tr[y][0]);
    tr[x][1]=merge(tr[x][1],tr[y][1]);
    return x;
}
int query(int x,int y,int d,int A){
    if(d<0)return A;
    int v=0;
    if(tr[x][0]&&tr[y][1]){
        v=max(v,query(tr[x][0],tr[y][1],d-1,A+(1<<d)));
    }
    if(tr[x][1]&&tr[y][0]){
        v=max(v,query(tr[x][1],tr[y][0],d-1,A+(1<<d)));
    }
    if(v)return v;
    if(tr[x][0]&&tr[y][0]){
        v=max(v,query(tr[x][0],tr[y][0],d-1,A));
    }
    if(tr[x][1]&&tr[y][1]){
        v=max(v,query(tr[x][1],tr[y][1],d-1,A));
    }
    return v;
}
void dfs(int k){
    for(int i=0;i<G[k].size();i++){
        dfs(G[k][i]);
        root[k]=merge(root[k],root[G[k][i]]);
        num[k]+=num[G[k][i]];
    }
    if(s[k].id!=0)add(root[k],w[s[k].id]),num[k]++;
    if(num[k]>1)ans=max(ans,query(root[k],root[k],20,0)+s[k].Max);
}
int main()
{    
    cin>>n;scanf("%s",ch+1);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    for(int i=n;i>=1;i--)ins(ch[i]-'a',i);
    for(int i=1;i<=cnt;i++)G[s[i].par].push_back(i);
    dfs(rt);
    cout<<ans<<endl;
    return 0;
}