[BZOJ][HAOI2015]T2（链剖+dfs序）

题目描述

传送门

题解

一上午就调些傻逼题，最近数据结构老是写挂心里很不爽。fye告诉我说多写写就不会错了。
傻逼链剖+dfs序，但是后来全改成LL才过了。也不知道为什么。学姐说LL和int必须强转，否则会出错。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long

const int max_n=1e5+5;
const int max_sum=max_n*5;
const int max_e=max_n*2;

LL n,m,x,y,opt,a,N;
LL w[max_n];
LL tot,point[max_n],v[max_e],next[max_e];
LL h[max_n],size[max_n],father[max_n],son[max_n];
LL in[max_n],out[max_n],top[max_n],num[max_n];
LL sum[max_sum],delta[max_sum];

inline void addedge(LL x,LL y){
    ++tot; next[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; next[tot]=point[y]; point[y]=tot; v[tot]=x;
}

inline void dfs_1(LL x,LL fa,LL dep){
    size[x]=1; h[x]=dep; father[x]=fa;
    LL maxson=0;
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        dfs_1(v[i],x,dep+1);
        size[x]+=size[v[i]];
        if (size[v[i]]>maxson){
            maxson=size[v[i]];
            son[x]=v[i];
          }
      }
}

inline void dfs_2(LL x,LL fa){
    if (son[fa]!=x) top[x]=x;
    else top[x]=top[fa];
    in[x]=++N;
    if (son[x]) dfs_2(son[x],x);
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa&&v[i]!=son[x])
        dfs_2(v[i],x);
    out[x]=N;
}

inline void update(LL now){
    sum[now]=sum[now<<1]+sum[now<<1|1];
}

inline void pushdown(LL now,LL l,LL r,LL mid){
    if (delta[now]){
        sum[now<<1]+=delta[now]*(mid-l+1);
        delta[now<<1]+=delta[now];
        sum[now<<1|1]+=delta[now]*(r-mid);
        delta[now<<1|1]+=delta[now];
        delta[now]=0;
    }
}

inline void build(LL now,LL l,LL r){
    LL mid=(l+r)>>1;
    if (l==r){
        sum[now]=w[num[l]];
        return;
    }
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}

inline void interval_change(LL now,LL l,LL r,LL lrange,LL rrange,LL v){
    LL mid=(l+r)>>1;
    if (lrange<=l&&r<=rrange){
        sum[now]+=v*(r-l+1);
        delta[now]+=v;
        return;
    }
    pushdown(now,l,r,mid);
    if (lrange<=mid)
      interval_change(now<<1,l,mid,lrange,rrange,v);
    if (mid+1<=rrange)
      interval_change(now<<1|1,mid+1,r,lrange,rrange,v);
    update(now);
}

inline LL query(LL now,LL l,LL r,LL lrange,LL rrange){
    LL mid=(l+r)>>1; LL ans=0;
    if (lrange<=l&&r<=rrange) return sum[now];
    pushdown(now,l,r,mid);
    if (lrange<=mid)
      ans+=query(now<<1,l,mid,lrange,rrange);
    if (mid+1<=rrange)
      ans+=query(now<<1|1,mid+1,r,lrange,rrange);
    return ans;
}



inline LL QUERY(LL u,LL t){
    LL f1=top[u],f2=top[t]; LL ans=0;
    while (f1!=f2){
        if (h[f1]<h[f2]){
            swap(f1,f2);
            swap(u,t);
        }
        ans+=query(1,1,N,in[f1],in[u]);
        u=father[f1];
        f1=top[u];
    }
    if (in[u]>in[t]) swap(u,t);
    ans+=query(1,1,N,in[u],in[t]);
    return ans;
}

int main(){
    scanf("%lld%lld",&n,&m);
    for (int i=1;i<=n;++i)
      scanf("%lld",&w[i]);
    for (int i=1;i<n;++i){
        scanf("%lld%lld",&x,&y);
        addedge(x,y);
    }

    dfs_1(1,0,1);
    dfs_2(1,0);
    for (int i=1;i<=n;++i)
      num[in[i]]=i;

    build(1,1,N);

    for (int i=1;i<=m;++i){
        scanf("%lld",&opt);
        switch(opt){
            case 1:{
                scanf("%lld%lld",&x,&a);
                interval_change(1,1,N,in[x],in[x],a);
                break;
            }
            case 2:{
                scanf("%lld%lld",&x,&a);
                interval_change(1,1,N,in[x],out[x],a);
                break;
            }
            case 3:{
                scanf("%lld",&x);
                LL ans=QUERY(1,x);
                printf("%lld\n",ans);
                break;
            }
        }
    }
}

总结

数据范围注意一下（强转什么的）。