bzoj4477: [Jsoi2015]字符串树(主席树+Hash+Lca)

2021/01/28 08:41
阅读数 32

题目传送门

解法:
长度小于等于10?
hash+主席树啊!!!
然后我就苦逼了的码了一个钟。
然后跑的贼慢倒数第三。。
网上的据说全是可持久化字典树??
什么鬼我不会。。
只能用主席树了。






建10棵主席树来维护长度不同的前缀。
然后树上主席树求一下就好了?

代码实现:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct node {
  
  int lc,rc,c;}t[11][2100000];int rt[11][110000],cnt;
void build(int c,int &u,int l,int r,int p) {
    if(u==0)u=++cnt;t[c][u].c++;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(c,t[c][u].lc,l,mid,p);
    else build(c,t[c][u].rc,mid+1,r,p);
}
void Merge(int c,int &u1,int u2) {
    if(u1==0){u1=u2;return ;}if(u2==0)return ;
    t[c][u1].c+=t[c][u2].c;
    Merge(c,t[c][u1].lc,t[c][u2].lc);
    Merge(c,t[c][u1].rc,t[c][u2].rc);
}
struct trnode {
  
  int x,y,next;}a[210000],e[110000];int len,last[110000];
char ss[110000][11];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int mx[21][110000],dep[110000],n;
void dfs(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]) {dep[y]=dep[x]+1;mx[0][y]=x;dfs(y);}
    }
}
unsigned int base=222;
int bin[21];
void work() {
    bin[0]=1;for(int i=1;i<=20;i++)bin[i]=bin[i-1]*2;
    for(int j=1;j<=20;j++)for(int i=1;i<=n;i++)if(dep[i]>=bin[j]) {
        mx[j][i]=mx[j-1][mx[j-1][i]];
    }
}
int lca(int x,int y) {
    if(dep[x]>dep[y])swap(x,y);
    for(int i=20;i>=0;i--)if(dep[y]-dep[x]>=bin[i])y=mx[i][y];
    if(x==y)return x;
    for(int i=20;i>=0;i--) if(dep[x]>=bin[i]&&mx[i][x]!=mx[i][y]) {
        x=mx[i][x];y=mx[i][y];
    }return mx[0][x];
}
int find(int c,int u1,int u2,int u3,int l,int r,int p) {
    if(t[c][u1].c+t[c][u2].c-2*t[c][u3].c==0)return 0;
    if(l==r)return t[c][u1].c+t[c][u2].c-2*t[c][u3].c;
    int mid=(l+r)/2;
    if(p<=mid)return find(c,t[c][u1].lc,t[c][u2].lc,t[c][u3].lc,l,mid,p);
    else return find(c,t[c][u1].rc,t[c][u2].rc,t[c][u3].rc,mid+1,r,p);
}
struct edge {
  
  int id,len,x;}S[1100000];
bool cmp(edge n1,edge n2) {
  
  return n1.x<n2.x;}
int s[1100000],tt=0;
void merge(int x) {
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=mx[0][x]) {
            for(int i=1;i<=10;i++)Merge(i,rt[i][y],rt[i][x]);
            merge(y);
        }
    }
}
int pos(int x) {
    int l=1,r=tt,mid,ans=-1;
    while(l<=r) {
        int mid=(l+r)/2;
        if(S[mid].x<=x) {
            if(S[mid].x==x)ans=mid;
            l=mid+1;
        }else r=mid-1;
    }return ans;
}char st[11];
int main() {
    scanf("%d",&n);len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++) {
        scanf("%d%d%s",&e[i].x,&e[i].y,ss[i]+1);
        ins(e[i].x,e[i].y);ins(e[i].y,e[i].x);
    }dep[1]=0;mx[0][1]=0;dfs(1);work();
    for(int i=1;i<n;i++)if(dep[e[i].x]>dep[e[i].y])swap(e[i].x,e[i].y);
    tt=0;
    for(int K=1;K<=10;K++) {

        for(int i=1;i<n;i++) {
            unsigned int ans=0;
            if(strlen(ss[i]+1)<K)continue;
            for(int j=1;j<=K;j++) ans=ans*base+ss[i][j]-'a'+1;
            S[++tt].x=ans;S[tt].id=i;S[tt].len=K;
        }
    }
    sort(S+1,S+1+tt,cmp);
    int tot=0;
    for(int i=1;i<=tt;i++) {
        if(S[i].x!=S[i-1].x)tot++;s[i]=tot;
    }
    for(int i=1;i<=10;i++) {
        cnt=0;
        for(int j=1;j<=tt;j++) {
            if(S[j].len!=i)continue;
            int k=S[j].id;
            build(i,rt[i][e[k].y],1,tot,s[j]);
        }
    }
    merge(1);
    int m;scanf("%d",&m);
    while(m--) {
        int x,y;scanf("%d%d%s",&x,&y,st+1);
        int T=strlen(st+1);
        unsigned int Hash=0;
        for(int i=1;i<=T;i++) Hash=Hash*base+st[i]-'a'+1;
        int X=pos(Hash);
        if(X==-1)printf("0\n");
        else {
            int Lca=lca(x,y);
            printf("%d\n",find(T,rt[T][x],rt[T][y],rt[T][Lca],1,tot,s[X]));
        }
    }
    return 0;
}
展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部