BJ模拟:Period on tree(Hash)

传送门

题解:
一个数 d d i循环节,则任何 (dj)|i ( d j ) | i 都是循环节,我们只需要试除每一个因子判断循环节是否为其约数即可。
判断用哈希+树状数组,注意预处理逆元,时间复杂度 O(nlog2n) O ( n log 2 ⁡ n )

#include <bits/stdc++.h>
#include <tr1/unordered_map>
using namespace std;
typedef long long LL;
typedef pair <int,int> pii;
typedef unsigned long long ULL;

const int RLEN=1<<18|1;
inline char nc() {
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
    char ch=nc(); int i=0,f=1;
    while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
    while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    return i*f;
}
inline void W(int x) {
    static int buf[50];
    if(!x) {putchar('0'); return;}
    if(x<0) {putchar('-'); x=-x;}
    while(x) {buf[++buf[0]]=x%10; x/=10;}
    while(buf[0]) {putchar(buf[buf[0]--]+'0');}
}

const int N=1e5+50;
const ULL base=31, invb=17256631552825064415;
int n, dep[N], top[N], fa[N], dfn[N], sze[N], son[N], ind;
int mnpr[N], pr[N], npr[N], pt;
ULL bitu[N], bitd[N], pw[N], iw[N], val[N];
vector <pii> edge[N];
vector <int> chain[N];
inline void inc(ULL *bit,int p,ULL v) {for(;p<=n;p+=(p&(-p))) bit[p]+=v;}
inline ULL ask(ULL *bit,int p,ULL v=0) {for(;p;p-=(p&(-p))) v+=bit[p]; return v;}
inline void dfs(int x,int f) {
    dep[x]=dep[f]+1; fa[x]=f; 
    sze[x]=1; dfn[x]=++ind;
    for(auto v:edge[x]) if(v.first!=f) {
        val[v.first]=v.second;
        dfs(v.first, x);
        sze[x]+=sze[v.first];
        son[x]=(sze[son[x]] < sze[v.first]) ? v.first : son[x];
    }
    ULL v1=pw[dep[x]]*val[x], v2=pw[n-dep[x]]*val[x];
    inc(bitu,dfn[x],v2); inc(bitu,dfn[x]+sze[x],-v2);
    inc(bitd,dfn[x],v1); inc(bitd,dfn[x]+sze[x],-v1);
}
inline void dfstop(int x,int f) {
    chain[top[x]].push_back(x);
    for(auto v:edge[x]) if(v.first!=f) {
        top[v.first]=(v.first==son[x]) ? top[x] : v.first;
        dfstop(v.first,x);
    }
} 
inline void init() {
    pw[0]=1; iw[0]=1;
    for(int i=1;i<=n;i++) pw[i]=pw[i-1]*base, iw[i]=iw[i-1]*invb;
    dfs(1,0); top[1]=1; dfstop(1,0);
    for(int i=2;i<=n;i++) {
        if(!npr[i]) pr[++pt]=i, mnpr[i]=i;
        for(int j=1;j<=pt;j++) {
            int k=i*pr[j];
            if(k>n) break;
            npr[k]=1; mnpr[k]=pr[j];
            if(!(i%pr[j])) break;
        }
    }
}
inline void modify(int x,int v) {
    ULL v1=pw[dep[x]]*val[x], v2=pw[n-dep[x]]*val[x];   
    inc(bitu,dfn[x],-v2); inc(bitu,dfn[x]+sze[x],v2);
    inc(bitd,dfn[x],-v1); inc(bitd,dfn[x]+sze[x],v1);
    val[x]=v, v1=pw[dep[x]]*v, v2=pw[n-dep[x]]*v;   
    inc(bitu,dfn[x],v2); inc(bitu,dfn[x]+sze[x],-v2);
    inc(bitd,dfn[x],v1); inc(bitd,dfn[x]+sze[x],-v1);
}
inline int lca(int x,int y) {
    while(top[x]!=top[y]) {
        (dep[top[x]] > dep[top[y]]) ? (x=fa[top[x]]) : (y=fa[top[y]]);
    } return (dep[x] > dep[y]) ? y : x;
}
inline int up(int x,int d) {
    while(dep[x]-dep[top[x]]<d) 
        d-=(dep[x]-dep[top[x]]+1), x=fa[top[x]];
    return chain[top[x]][dep[x]-dep[top[x]]-d];
}
inline ULL gu(int x,int f) {
    return (ask(bitu,dfn[x])-ask(bitu,dfn[f]))*iw[n-dep[x]];
}
inline ULL gd(int x,int f) {
    return (ask(bitd,dfn[x])-ask(bitd,dfn[f]))*iw[dep[f]+1];
}
inline ULL gv1(int u,int l,int v,int len) {
    int d1=min(dep[u]-dep[l], len), u2=up(u,d1);
    ULL V1=gu(u,u2);
    if(d1<len) {
        int d2=len-d1, v2=up(v,dep[v]-dep[l]-d2);
        ULL V2=gd(v2,l);
        return V1+V2*pw[d1];
    } return V1;
}
inline ULL gv2(int u,int l,int v,int len) {
    int d1=min(dep[v]-dep[l], len), v2=up(v,d1);
    ULL V1=gd(v,v2);
    if(d1<len) {
        int d2=len-d1, u2=up(u,dep[u]-dep[l]-d2);
        ULL V2=gu(u2,l);
        return V2+V1*pw[d2];
    } return V1;
}
inline int ask(int u,int v,int l) {
    int len=dep[u]+dep[v]-2*dep[l];
    vector <int> fac; int lst=0,x=len;
    while(x!=1) {
        if(mnpr[x]!=lst)
            fac.push_back(mnpr[x]),lst=mnpr[x];
        x/=mnpr[x];
    } x=len;
    for(int j=0;j<fac.size();++j) {
        while(!(x%fac[j])) {
            int tp=x/fac[j];
            if(gv1(u,l,v,len-tp)==gv2(u,l,v,len-tp)) x=tp;
            else break;
        }
    } return x;
}
int main() {
    n=rd();
    for(int i=1;i<n;i++) {
        int x=rd(), y=rd(), w=nc()-'a';
        edge[x].push_back(pii(y,w));
        edge[y].push_back(pii(x,w));
    }
    init();
    for(int i=rd();i;i--) {
        int op=rd();
        if(op==1) {
            int u=rd(), v=rd(), l=lca(u,v);
            W(ask(u,v,l)); putchar('\n');
        } else {
            int u=rd(), v=rd(); 
            if(dep[u]>dep[v]) swap(u,v);
            modify(v,nc()-'a');
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值