bzoj3091 城市旅行(lct+数学)

记链x->y的长度为n,则答案的分母显然就是n*(n+1)/2,我们的问题在于维护分子。考虑每一个点对答案的贡献,就是 v[i]i(ni+1) 这个东西怎么维护呢?我们维护 ls=v[i]i,rs=v[i](ni+1) 即一个左起排名乘点权,一个右起排名乘点权。这两个东西怎么维护呢?ls[x]=ls[l]+ls[r]+(sum[r]+v[x])*(sz[l]+1)。因为我们只是右子树上的点及x的排名增大了(sz[l]+1),把这个差值加上即可。rs同理的维护。那么我们就可以根据这两个值得出答案as。
add的时候怎么维护呢?推一推公式就好了-。-
注意这个rev要即时生效,因为ls,rs的存在。

因为read出锅wa了一上午,真是悲惨的经历。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define inf 0x3f3f3f3f
#define N 50010
#define ll long long
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,fa[N],c[N][2],q[N];
ll ls[N],rs[N],as[N],sum[N],v[N],sz[N],add[N];bool rev[N];
inline bool isroot(int x){
    return x!=c[fa[x]][0]&&x!=c[fa[x]][1];
}
inline void update(int x){
    int l=c[x][0],r=c[x][1];
    sz[x]=sz[l]+sz[r]+1;sum[x]=sum[l]+sum[r]+v[x];
    ls[x]=ls[l]+ls[r]+(v[x]+sum[r])*(sz[l]+1);
    rs[x]=rs[r]+rs[l]+(v[x]+sum[l])*(sz[r]+1);
    as[x]=as[l]+ls[l]*(sz[r]+1)+as[r]+rs[r]*(sz[l]+1)+v[x]*(sz[l]+1)*(sz[r]+1);
}
inline void dorev(int x){
    if(!x) return;
    rev[x]^=1;swap(c[x][0],c[x][1]);
    swap(ls[x],rs[x]);
}
inline void doadd(int x,ll val){
    if(!x||!val) return;
    add[x]+=val;v[x]+=val;sum[x]+=val*sz[x];
    ls[x]+=val*(sz[x]+1)*sz[x]/2;
    rs[x]+=val*(sz[x]+1)*sz[x]/2;
    as[x]+=val*sz[x]*(sz[x]+1)*(sz[x]+2)/6;
}
inline void pushdown(int x){
    if(rev[x]) rev[x]=0,dorev(c[x][0]),dorev(c[x][1]);
    if(add[x]) doadd(c[x][0],add[x]),doadd(c[x][1],add[x]),add[x]=0;
}
inline void rotate(int x){
    int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
    if(!isroot(y)) c[z][y==c[z][1]]=x;
    fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
    c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x){
    int top=0;q[++top]=x;
    for(int xx=x;!isroot(xx);xx=fa[xx]) q[++top]=fa[xx];
    while(top) pushdown(q[top--]);
    while(!isroot(x)){
        int y=fa[x],z=fa[y];
        if(!isroot(y)){
            if(x==c[y][1]^y==c[z][1]) rotate(x);
            else rotate(y);
        }rotate(x);
    }
}
inline void access(int x){
    int y=0;
    while(x){splay(x);c[x][1]=y;update(x);y=x;x=fa[x];}
}
inline void makeroot(int x){
    access(x);splay(x);dorev(x);
}
inline int find(int x){
    access(x);splay(x);while(c[x][0]) x=c[x][0];return x;
}
inline void link(int x,int y){
    makeroot(x);fa[x]=y;
}
inline void cut(int x,int y){
    if(find(x)!=find(y)) return;
    makeroot(x);access(y);splay(y);
    if(c[y][0]==x) c[y][0]=fa[x]=0,update(y);
}
inline void add1(int x,int y,ll val){
    makeroot(x);access(y);splay(y);doadd(y,val);
}
inline ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
inline void ask(int x,int y){
    if(find(x)!=find(y)){puts("-1");return;}
    makeroot(x);access(y);splay(y);
    ll a=as[y],b=sz[y]*(sz[y]+1)/2;
    ll g=gcd(a,b);
    printf("%lld/%lld\n",a/g,b/g);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=n;++i) v[i]=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();link(x,y);
    }while(m--){
        int op=read(),x=read(),y=read();
        if(op==1) cut(x,y);
        else if(op==2){if(find(x)!=find(y)) link(x,y);}
        else if(op==3){int val=read();if(find(x)==find(y)) add1(x,y,val);}
        else ask(x,y);
    }return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值