bzoj4928: 第二题

Description

对于一棵有根树，定义一个点u的k-子树为u的子树中距离u不超过k的部分。

注意，假如u的子树中不存在距离u为k的点，则u的k-子树是不存在的。

定义两棵子树是相同的，当且仅当不考虑点的标号时，他们的形态是

相同的（儿子的顺序也需要考虑）。给定一棵n个点，点的标号在[1,n]，

以1为根的有根树。问最大的k，使得存在两个点u !=v，满足u的k-子树与v的k-子树相同。

Input

第一行输入一个正整数n。

接下来读入n个部分，第i个部分描述点i的儿子，且以顺序给出。

每个部分首先读入一个整数x，代表儿子个数。

接下来x个整数，代表从左到右儿子的标号

n ≤ 100000，保证给出的树是合法的

Output

输出一个整数k，代表最大的合法的k

已知每个点的a-子树和b-子树的hash，可以推出每个点的(a+b)-子树的hash

对答案倍增，hash判定

#include<cstdio>
#include<cstring>
#define Fe(i,l,r) for(int i=l;i<=r;++i)
typedef unsigned long long u64;
const int N=,P=;
void maxs(int&a,int b){if(a<b)a=b;}
u64 H(u64 x){return (x^x>>^x<<)+;}
char buf[],*ptr=buf+;
int G(){
    if(ptr-buf==)fread(ptr=buf,,,stdin);
    return *ptr++;
}
int _(){
    int x=;
    if(ptr-buf<){
        while(*ptr<)++ptr;
        while(*ptr>)x=x*+*ptr++-;
    }else{
        int c=G();
        while(c<)c=G();
        while(c>)x=x*+c-,c=G();
    }
    return x;
}
int n,ans=;
int mem[N],*mp=mem,*e[N],ep[N],fa[][N],fas[N],id[N],_fa[N],_son[N];
int q[N],ql=,qr=,dep[N],md[N];
u64 h[][N],h2[N],_h[N];
u64 hx[P];
int ht[P],tk=;
bool ins(u64 x){
    int w=(unsigned(x)^unsigned(x>>))%P;
    while(ht[w]==tk){
        if(hx[w]==x)return ;
        if((w+=)>=P)w-=P;
    }
    hx[w]=x;ht[w]=tk;
    return ;
}
template<class T>
void cpy(T*a,T*b){memcpy(a+,b+,sizeof(T)*n);}
void getson(){
    memset(_son+,,sizeof(int)*n);
    Fe(t,,qr){
        int w=q[t],f=fas[w];
        if(!_son[f])_son[f]=t;
    }
}
void cal(int _i){
    cpy(_h,h[_i]);
    cpy(_fa,fa[_i]);
    for(int i=_i-;i>=;--i){
        ++tk;
        cpy(h2,_h);
        getson();
        Fe(t,,qr){
            int w=q[t],f=_fa[w];
            if(f)h2[f]=h2[f]*+h[i][w]+(id[fa[][w]]-_son[f]);
        }
        bool is=;
        int mn=ans+(<<i);
        Fe(w,,n)if(md[w]+>=mn&&ins(h2[w])){
            is=;
            break;
        }
        if(is){
            ans=mn;
            Fe(w,,n){
                _h[w]=H(h2[w]);
                _fa[w]=fa[i][_fa[w]];
                fas[w]=fa[i][fas[w]];
            }
        }
    }
}
void cal(){
    Fe(w,,n)h[][w]=;
    for(int i=;i<;++i){
        ++tk;
        cpy(h2,h[i]);
        getson();
        Fe(t,,qr){
            int w=q[t],f=fa[i][w];
            if(f)h2[f]=h2[f]*+h[i][w]+(id[fa[][w]]-_son[f]);
        }
        bool is=;
        int mn=ans+(<<i);
        Fe(w,,n)if(md[w]+>=mn&&ins(h2[w])){
            is=;
            break;
        }
        if(is){
            ans=mn;
            Fe(w,,n){
                h[i+][w]=H(h2[w]);
                fa[i+][w]=fa[i][fa[i][w]];
                fas[w]=fa[i][fas[w]];
            }
        }else return cal(i);
    }
}
int main(){
    n=_();
    for(int i=;i<=n;++i){
        fas[i]=i;
        e[i]=mp;
        mp+=ep[i]=_();
        for(int j=;j<ep[i];++j)fa[][e[i][j]=_()]=i;
    }
    for(q[++qr]=;fa[][q[]];++q[]);
    while(ql!=qr){
        int w=q[++ql];
        id[w]=ql;
        for(int*l=e[w],*r=l+ep[w];l!=r;++l)dep[q[++qr]=*l]=dep[w]+;
    }
    for(int i=qr;i;--i){
        int w=q[i];
        maxs(md[fa[][w]],md[w]+);
    }
    cal();
    printf("%d\n",ans-);
    return ;
}