bzoj3439: Kpm的MC密碼(主席樹+DFS序+字典樹)

Hanks_o發表於2018-04-18

題目傳送門
做這題有人跟我說用連結串列。處理相同的串。
網上都說要。。
其實不用吧。。記錄每個串的結尾是在字典樹上哪個點就行啊。
然後一個一個插啊。

解法:
因為是字尾所以到這建字典樹。
然後kpm串肯定是子樹的所有串。
那麼用主席樹維護子樹第k小。
要求編號連續就套個dfs序就行了。

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct Trie {int son[31];Trie() {memset(son,-1,sizeof(son));}}tr[310000];
int tot,end[110000],root;char ss[310000];
void Build(int k) {
    int x=root,len=strlen(ss+1);
    for(int i=len;i>=1;i--) {
        int y=ss[i]-'a'+1;
        if(tr[x].son[y]==-1) {tot++;tr[x].son[y]=tot;}
        x=tr[x].son[y];
    }
    end[k]=x;
}
struct node {int lc,rc,c;}t[2100000];int cnt,rt[110000];
void build(int &u,int l,int r,int p,int c) {
    if(u==0)u=++cnt;t[u].c+=c;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p,c);
    else build(t[u].rc,mid+1,r,p,c);
}
void Merge(int &u1,int u2) {
    if(u1==0){u1=u2;return ;}if(u2==0)return ;
    t[u1].c+=t[u2].c;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
int z,st[310000],ed[310000];
void dfs(int x) {
    st[x]=++z;
    for(int i=1;i<=26;i++)if(tr[x].son[i]!=-1)dfs(tr[x].son[i]);
    ed[x]=z;
}
int find(int u1,int u2,int l,int r,int k) {
    if(t[u1].c-t[u2].c<k)return -1;
    if(l==r)return l;int mid=(l+r)/2;
    int c=t[t[u1].lc].c-t[t[u2].lc].c;
    if(k<=c)return find(t[u1].lc,t[u2].lc,l,mid,k);
    else return find(t[u1].rc,t[u2].rc,mid+1,r,k-c);
}
int find_l(int u1,int u2,int l,int r,int p) {
    if(l==r)return t[u1].c-t[u2].c;int mid=(l+r)/2;
    int c=t[t[u1].lc].c-t[t[u2].lc].c;
    if(p<=mid)return find(t[u1].lc,t[u2].lc,l,mid,p);
    else return find(t[u1].rc,t[u2].rc,mid+1,r,p)+c;
}
int main() {
    int n;scanf("%d",&n);root=tot=0;
    for(int i=1;i<=n;i++) {scanf("%s",ss+1);Build(i);}
    z=0;dfs(root);cnt=0;
    for(int i=1;i<=n;i++)build(rt[st[end[i]]],1,n,i,1);
    for(int i=1;i<=z;i++)Merge(rt[i],rt[i-1]);
    for(int i=1;i<=n;i++) {
        int x,k;scanf("%d",&k);x=end[i];printf("%d\n",find(rt[ed[x]],rt[st[x]-1],1,n,k));
    }
    return 0;
}

相關文章