BZOJ3091: 城市旅行

Description

Input

Output

Sample Input

4 5
1 3 2 5
1 2
1 3
2 4
4 2 4
1 2 4
2 3 4
3 1 4 1
4 1 4

Sample Output

16/3
6/1

HINT

对于所有数据满足 1<=N<=50,000 1<=M<=50,000 1<=Ai<=10^6 1<=D<=100 1<=U,V<=N

 

恶心的动态树上维护各种信息。

不难发现ans=ΣAi*i*(len-i+1)。

我们在splay树上维护几个值：
sumv=ΣAi*i*(len-i+1)

lsum=ΣAi*i

rsum=ΣAi*(len-i+1)

sum=ΣAi

不难维护者几个变量的关系，打懒标记时快速算一下Σi*(len-i+1)和Σi就行了。

注意flip时要交换两子树的lsum和rsum。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cctype>
#define lc ch[x][0]
#define rc ch[x][1]
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i!=-1;i=next[i])
using namespace std;
const int BufferSize=<<;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
    if(head==tail) {
        int l=fread(buffer,,BufferSize,stdin);
        tail=(head=buffer)+l;
    }
    return *head++;
}
inline int read() {
    int x=,f=;char c=Getchar();
    for(;!isdigit(c);c=Getchar()) if(c=='-') f=-;
    for(;isdigit(c);c=Getchar()) x=x*+c-'';
    return x*f;
}
typedef long long ll;
const int maxn=;
ll f[maxn],sumv[maxn],sum[maxn],lsum[maxn],rsum[maxn],s[maxn],addv[maxn],val[maxn];
int pre[maxn],fa[maxn],ch[maxn][],flip[maxn];
void Add(int x,ll v) {
    if(!v||!x) return;
    sumv[x]+=v*f[s[x]];
    lsum[x]+=v*(+s[x])*s[x]/;
    rsum[x]+=v*(+s[x])*s[x]/;
    sum[x]+=v*s[x];
}
void maintain(int x) {
    if(!x) return;
    s[x]=s[lc]+s[rc]+;
    sum[x]=sum[lc]+sum[rc]+val[x];
    sumv[x]=sumv[lc]+sumv[rc]+lsum[lc]*(s[rc]+)+rsum[rc]*(s[lc]+)+val[x]*(s[lc]+)*(s[rc]+);
    lsum[x]=lsum[lc]+lsum[rc]+val[x]*(s[lc]+)+sum[rc]*(s[lc]+);
    rsum[x]=rsum[rc]+rsum[lc]+val[x]*(s[rc]+)+sum[lc]*(s[rc]+);
    Add(x,addv[x]);
}
void pushdown(int x) {
    if(flip[x]) {
        swap(lc,rc);
        swap(lsum[lc],rsum[lc]);
        swap(lsum[rc],rsum[rc]);
        flip[lc]^=;flip[rc]^=;
        flip[x]=;
    }
    if(addv[x]) {
        val[x]+=addv[x];addv[lc]+=addv[x];addv[rc]+=addv[x];
        Add(lc,addv[x]);Add(rc,addv[x]);addv[x]=;
    }
}
void rotate(int x) {
    int y=pre[x],z=pre[y],d=ch[y][]==x;
    ch[y][d^]=ch[x][d];pre[ch[x][d]]=y;
    ch[z][ch[z][]==y]=x;pre[x]=z;
    ch[x][d]=y;pre[y]=x;maintain(y);
}
int S[maxn],top;
void splay(int x) {
    for(int i=x;i;i=pre[i]) S[++top]=i;
    if(top!=) fa[x]=fa[S[top]],fa[S[top]]=;
    while(top) pushdown(S[top--]);
    while(pre[x]) rotate(x);
    maintain(x);
}
void access(int x) {
    for(int y=;x;x=fa[x]) {
        splay(x);pre[ch[x][]]=;fa[ch[x][]]=x;
        ch[x][]=y;pre[y]=x;maintain(y=x);
    }
}
void makeroot(int x) {
    access(x);splay(x);flip[x]^=;
    maintain(x);
}
int find(int x) {
    access(x);splay(x);
    while(ch[x][]) x=ch[x][];
    return x;
}
void link(int x,int y) {
    makeroot(x);fa[x]=y;
}
void cut(int x,int y) {
    makeroot(x);access(y);splay(y);
    if(s[y]==) {
        pre[ch[y][]]=;ch[y][]=;
        maintain(y);
    }
}
ll gcd(ll x,ll y) {return !y?x:gcd(y,x%y);}
void query(int x,int y) {
    makeroot(x);access(y);splay(y);
    ll len=s[y]-;len=(len+)*(len+)/;
    ll ans=sumv[y],t=gcd(len,ans);
    printf("%lld/%lld\n",ans/t,len/t);
}
void update(int x,int y,ll v) {
    makeroot(x);access(y);splay(y);
    addv[y]+=v;Add(y,v);
}
int main() {
    int n=read(),m=read();
    f[]=;
    rep(i,,n) f[i]=f[i-]+(ll)i*(i-)/+i;
    rep(i,,n) val[i]=read();
    rep(i,,n-) link(read(),read());
    while(m--) {
        int t=read(),x=read(),y=read();
        if(t==) if(find(x)==find(y)) cut(x,y);
        if(t==) if(find(x)!=find(y)) link(x,y);
        if(t==) {
            ll v=read();
            if(find(x)==find(y)) update(x,y,v);
        }
        if(t==) {
            if(find(x)==find(y)) query(x,y);
            else puts("-1");
        }
    }
    return ;
}