[BZOJ4034][HAOI2015]树上操作（链剖+dfs序）

题目描述

传送门

题解

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

代码

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

int n,m,opt,x,y,z,dfs_clock;
int tot,point[N],nxt[N*2],v[N*2];
int a[N],num[N],father[N],son[N],size[N],h[N],in[N],out[N],top[N];
LL sum[N*4],delta[N*4],ans;

void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void dfs_1(int x,int fa)
{
    father[x]=fa;size[x]=1;h[x]=h[fa]+1;
    for (int i=point[x];i;i=nxt[i])
        if (v[i]!=fa)
        {
            dfs_1(v[i],x);
            size[x]+=size[v[i]];
            if (size[v[i]]>size[son[x]]) son[x]=v[i];
        }
}
void dfs_2(int x,int fa)
{
    if (son[fa]==x) top[x]=top[fa];
    else top[x]=x;
    in[x]=++dfs_clock;num[dfs_clock]=a[x];
    if (son[x]) dfs_2(son[x],x);
    for (int i=point[x];i;i=nxt[i])
        if (v[i]!=fa&&v[i]!=son[x])
            dfs_2(v[i],x);
    out[x]=dfs_clock;
}
void update(int now)
{
    sum[now]=sum[now<<1]+sum[now<<1|1];
}
void build(int now,int l,int r)
{
    int mid=(l+r)>>1;
    if (l==r)
    {
        sum[now]=(LL)num[l];
        return;
    }
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}
void pushdown(int now,int l,int r,int mid)
{
    if (delta[now])
    {
        sum[now<<1]+=delta[now]*(LL)(mid-l+1);delta[now<<1]+=delta[now];
        sum[now<<1|1]+=delta[now]*(LL)(r-mid);delta[now<<1|1]+=delta[now];
        delta[now]=0LL;
    }
}
void change(int now,int l,int r,int lr,int rr,int v)
{
    int mid=(l+r)>>1;
    if (lr<=l&&r<=rr)
    {
        sum[now]+=(LL)v*(r-l+1);
        delta[now]+=(LL)v;
        return;
    }
    pushdown(now,l,r,mid);
    if (lr<=mid) change(now<<1,l,mid,lr,rr,v);
    if (mid+1<=rr) change(now<<1|1,mid+1,r,lr,rr,v);
    update(now);
}
LL query(int now,int l,int r,int lr,int rr)
{
    int mid=(l+r)>>1;
    LL ans=0LL;
    if (lr<=l&&r<=rr) return sum[now];
    pushdown(now,l,r,mid);
    if (lr<=mid) ans+=query(now<<1,l,mid,lr,rr);
    if (mid+1<=rr) ans+=query(now<<1|1,mid+1,r,lr,rr);
    return ans;
}
LL ask(int u,int t)
{
    int f1=top[u],f2=top[t];
    LL ans=0LL;
    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("%d%d",&n,&m);
    for (int i=1;i<=n;++i) scanf("%d",&a[i]);
    for (int i=1;i<n;++i)
    {
        scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    dfs_1(1,0);
    dfs_2(1,0);
    build(1,1,n);
    for (int i=1;i<=m;++i)
    {
        scanf("%d",&opt);
        if (opt==1)
        {
            scanf("%d%d",&x,&z);
            change(1,1,n,in[x],in[x],z);
        }
        if (opt==2)
        {
            scanf("%d%d",&x,&z);
            change(1,1,n,in[x],out[x],z);
        }
        if (opt==3)
        {
            scanf("%d",&x);
            ans=ask(1,x);
            printf("%I64d\n",ans);
        }
    }
}

手工栈

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

int n,m,opt,x,y,z,dfs_clock;
int tot,point[N],nxt[N*2],v[N*2];
int a[N],num[N],father[N],son[N],size[N],h[N],in[N],out[N],top[N];
int stack[N],cur[N];bool use[N];
LL sum[N*4],delta[N*4],ans;

void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void dfs_1()
{
    int tmp=0;stack[++tmp]=1;
    size[1]=1;h[1]=1;
    for (int i=1;i<=n;++i) cur[i]=point[i];
    while (tmp)
    {
        int x=stack[tmp];
        while (cur[x]&&v[cur[x]]==father[x]) cur[x]=nxt[cur[x]];
        if (!cur[x])
        {
            --tmp;
            if (father[x])
            {
                size[father[x]]+=size[x];
                if (size[x]>size[son[father[x]]]) son[father[x]]=x;
            }
            continue;
        }
        int vt=v[cur[x]];stack[++tmp]=vt;
        father[vt]=x;size[vt]=1;h[vt]=h[x]+1;
        cur[x]=nxt[cur[x]];
    }
}
void dfs_2()
{
    int tmp=0;stack[++tmp]=1;
    top[1]=1;in[1]=++dfs_clock;num[dfs_clock]=a[1];
    for (int i=1;i<=n;++i) cur[i]=point[i];
    while (tmp)
    {
        int x=stack[tmp];
        if (!use[x])
        {
            use[x]=1;
            if (son[x])
            {
                int vt=son[x];
                top[vt]=top[x];in[vt]=++dfs_clock;num[dfs_clock]=a[vt];
                stack[++tmp]=vt;
            }
            continue;
        }
        while (cur[x]&&(v[cur[x]]==father[x]||v[cur[x]]==son[x])) cur[x]=nxt[cur[x]];
        if (!cur[x])
        {
            --tmp;
            out[x]=dfs_clock;
            continue;
        }
        int vt=v[cur[x]];stack[++tmp]=vt;
        top[vt]=vt;in[vt]=++dfs_clock;num[dfs_clock]=a[vt];
        cur[x]=nxt[cur[x]];
    }
}
void update(int now)
{
    sum[now]=sum[now<<1]+sum[now<<1|1];
}
void build(int now,int l,int r)
{
    int mid=(l+r)>>1;
    if (l==r)
    {
        sum[now]=(LL)num[l];
        return;
    }
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}
void pushdown(int now,int l,int r,int mid)
{
    if (delta[now])
    {
        sum[now<<1]+=delta[now]*(LL)(mid-l+1);delta[now<<1]+=delta[now];
        sum[now<<1|1]+=delta[now]*(LL)(r-mid);delta[now<<1|1]+=delta[now];
        delta[now]=0LL;
    }
}
void change(int now,int l,int r,int lr,int rr,int v)
{
    int mid=(l+r)>>1;
    if (lr<=l&&r<=rr)
    {
        sum[now]+=(LL)v*(r-l+1);
        delta[now]+=(LL)v;
        return;
    }
    pushdown(now,l,r,mid);
    if (lr<=mid) change(now<<1,l,mid,lr,rr,v);
    if (mid+1<=rr) change(now<<1|1,mid+1,r,lr,rr,v);
    update(now);
}
LL query(int now,int l,int r,int lr,int rr)
{
    int mid=(l+r)>>1;
    LL ans=0LL;
    if (lr<=l&&r<=rr) return sum[now];
    pushdown(now,l,r,mid);
    if (lr<=mid) ans+=query(now<<1,l,mid,lr,rr);
    if (mid+1<=rr) ans+=query(now<<1|1,mid+1,r,lr,rr);
    return ans;
}
LL ask(int u,int t)
{
    int f1=top[u],f2=top[t];
    LL ans=0LL;
    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()
{
    freopen("input.in","r",stdin);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i) scanf("%d",&a[i]);
    for (int i=1;i<n;++i)
    {
        scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    dfs_1();
    dfs_2();
    build(1,1,n);
    for (int i=1;i<=m;++i)
    {
        scanf("%d",&opt);
        if (opt==1)
        {
            scanf("%d%d",&x,&z);
            change(1,1,n,in[x],in[x],z);
        }
        if (opt==2)
        {
            scanf("%d%d",&x,&z);
            change(1,1,n,in[x],out[x],z);
        }
        if (opt==3)
        {
            scanf("%d",&x);
            ans=ask(1,x);
            printf("%I64d\n",ans);
        }
    }
}

总结

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