BJ模拟 Period on tree【树状数组+哈希】

题目描述：

给定一棵 N 个节点的无根树，每条边上有一个小写英文字母。每次我们选择两个不同的节点 u 和 v，然后依次写下从 u 到 v 的最短路径上每条边上的字母，我们就能得到这条路径对应的字符串。

我们称字符串 t 是另一个字符串 s 的循环，当且仅当 s 可由若干份的 t 串拼接而成。 你的任务，则是对于给定的树，处理下面两种操作：

• 1 u v，输出从 u 到 v 路径的字符串的最短循环的长度（注：数据保证 u≠v ）；

• 2 u v c，将 u 和 v 之间相连的边上的字母改为 c （注：数据保证此操作中u和v相邻）。

解题思路：

用dfs序的树状数组结合节点深度维护hash值。

询问时先求出链长len，那么循环节一定是len的约数，又注意到如果x不是循环节长度，那么x的约数肯定不是循环节，所以可以将len分解质因数后用试除法。

线性筛出每个数的最小质因数，分解质因数就是$O(logn)$的了。

判断长度x是否是循环节只需看hash(1,n-x)是否等于hash(x+1,n)即可。

时间复杂度$O(nlog^2n)$

#include<bits/stdc++.h>
#define ll unsigned long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();c!='-'&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=200005;
const ll base=123LL,inv_base=3449391168254631603LL;
int n,m,yz[N];
int tot,first[N],nxt[N<<1],to[N<<1],w[N<<1];
int idx,in[N],out[N],c[N],dep[N],fa[N][20];
int pn,pri[N],minpri[N];
ll bin[N],binv[N],bitup[N],bitdw[N];
void sieve()
{
    bin[0]=binv[0]=1;
    for(int i=1;i<=n;i++)bin[i]=bin[i-1]*base,binv[i]=binv[i-1]*inv_base;
    for(int i=2;i<=n;i++)
    {
        if(!minpri[i])minpri[i]=pri[++pn]=i;
        for(int j=1;j<=pn;j++)
        {
            ll k=i*pri[j];
            if(k>n)break; 
            minpri[k]=pri[j];
            if(i%pri[j]==0)break;
        }
    }
}
void add(int x,int y,int z)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y,w[tot]=z;
}
void modify(ll *bit,int i,ll v)
{
    for(;i<=n;i+=i&-i)bit[i]+=v;
}
ll query(ll *bit,int i)
{
    ll res=0;
    for(;i;i-=i&-i)res+=bit[i];
    return res;
}
void modify(int u,int ch)
{
    modify(bitup,in[u],bin[dep[u]]*ch),modify(bitup,out[u]+1,-bin[dep[u]]*ch);
    modify(bitdw,in[u],bin[n-dep[u]]*ch),modify(bitdw,out[u]+1,-bin[n-dep[u]]*ch);
}
void dfs(int u)
{
    in[u]=++idx;
    for(int i=1;i<20;i++)fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];
        if(v==fa[u][0])continue;
        fa[v][0]=u,dep[v]=dep[u]+1;
        c[v]=w[e],dfs(v);
    }
    out[u]=idx;modify(u,c[u]);
}
int jump(int x,int det)
{
    for(int i=0;(1<<i)<=det;i++)
        if(det>>i&1)x=fa[x][i];
    return x;
}
int lca(int x,int y)
{
    if(dep[x]<dep[y])swap(x,y);
    x=jump(x,dep[x]-dep[y]);if(x==y)return x;
    for(int i=19;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
ll getup(int x,int f){return (query(bitup,in[x])-query(bitup,in[f]))*binv[dep[f]+1];}
ll getdw(int x,int f){return (query(bitdw,in[x])-query(bitdw,in[f]))*binv[n-dep[x]];}
ll Hashup(int u,int v,int t,int len)
{
    int d1=min(len,dep[u]-dep[t]),u1=jump(u,d1);
    ll V1=getup(u,u1);
    if(d1<len)
    {
        int d2=len-d1,v2=jump(v,dep[v]-dep[t]-d2);
        ll V2=getdw(v2,t);V1=V1*bin[d2]+V2;
    }
    return V1;
}
ll Hashdw(int u,int v,int t,int len)
{
    int d1=min(len,dep[v]-dep[t]),v1=jump(v,d1);
    ll V1=getdw(v,v1);
    if(d1<len)
    {
        int d2=len-d1,u2=jump(u,dep[u]-dep[t]-d2);
        ll V2=getup(u2,t);V1=V1+V2*bin[d1];
    }
    return V1;
}
int query(int x,int y)
{
    int t=lca(x,y),len=dep[x]+dep[y]-2*dep[t],cnt=0,tmp=len,res=len;
    while(tmp!=1)yz[++cnt]=minpri[tmp],tmp/=minpri[tmp];
    for(int i=1;i<=cnt;i++)
    {
        tmp=res/yz[i];
        if(Hashup(x,y,t,len-tmp)==Hashdw(x,y,t,len-tmp))res=tmp;
        else while(i<cnt&&yz[i]==yz[i+1])i++;
    }
    return res;
}
int main()
{
    //freopen("lx.in","r",stdin);
    n=getint();sieve();
    int op,x,y,z;
    for(int i=1;i<n;i++)
    {
        x=getint(),y=getint(),z=getchar()-'a';
        add(x,y,z),add(y,x,z);
    }
    dfs(1);
    m=getint();
    while(m--)
    {
        op=getint(),x=getint(),y=getint();
        if(op==1)printf("%d\n",query(x,y));
        else x=dep[x]>dep[y]?x:y,z=getchar()-'a',modify(x,z-c[x]),c[x]=z;
    }
    return 0;
}