题目描述:
给定一棵 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) O ( l o g n ) 的了。
判断长度x是否是循环节只需看hash(1,n-x)是否等于hash(x+1,n)即可。
时间复杂度 O(nlog2n) O ( n l o g 2 n )
#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;
}