HAOI[2015]树上操作

59 篇文章 0 订阅
22 篇文章 0 订阅

树链剖分裸题。
需要注意的是,此题要开long long
中间变量也要开long long (坑死了)
话说自己搞到组数据,结果死活在自己电脑上RE。
后来发现是爆栈了。
下篇博客发一下本地扩栈的方法。
题目

#include <cstdio>
#include <iostream>
#define lson (o<<1),l,mid
#define rson (o<<1)+1,mid+1,r
#define ll long long
using namespace std;
const int maxm=210000;
int deep[maxm],son[maxm],size[maxm],fa[maxm],top[maxm];
int head[maxm],to[maxm*2],net[maxm*2],cnt;
ll p[maxm],val[maxm];
bool vis[maxm];
int id[maxm],tot;
int n,m,root;
struct node{
    int l,r;
    ll siz,sum,add;
};
node t[maxm*4];
inline void add(int x,int y)
{
    to[++cnt]=y;
    net[cnt]=head[x];
    head[x]=cnt;
}
void dfs1(int now,int fax,int dep)
{
    fa[now]=fax,deep[now]=dep,size[now]=1;
    int maxson=-1;
    for(int i=head[now];i;i=net[i])
    {
        if(to[i]==fax) return;
        dfs1(to[i],now,dep+1);
        size[now]+=size[to[i]];
        if(maxson<size[to[i]])
         son[now]=to[i],maxson=size[to[i]];
    }
}
void dfs2(int now,int topx)
{
    id[now]=++tot;
    val[tot]=p[now];
    top[now]=topx;
    if(!son[now]) return;
    dfs2(son[now],topx);
    for(int i=head[now];i;i=net[i])
     if(!id[to[i]])
      dfs2(to[i],to[i]);
}
inline void update(int o)
{
    t[o].sum=(t[(o<<1)].sum+t[(o<<1)|1].sum);
}
inline void pushdown(int o)
{
    ll adi=t[o].add;
    for(int i=0;i<=1;i++)
     t[(o<<1)+i].sum=(t[(o<<1)+i].sum+(t[(o<<1)+i].siz*adi)),t[(o<<1)+i].add=(t[(o<<1)+i].add+adi);
    t[o].add=0;
}
void adj(int o,int ql,int qr,ll num)
{
    int l=t[o].l,r=t[o].r;
    if(ql<=l&&qr>=r)
    {
        t[o].sum=(t[o].sum+(t[o].siz*num));
        t[o].add=(t[o].add+num);
        return;
    }
    pushdown(o);
    int mid=(l+r)>>1;
    if(ql<=mid) adj((o<<1),ql,qr,num);
    if(qr>mid) adj((o<<1)|1,ql,qr,num);
    update(o);
}
ll ask(int o,int ql,int qr)
{
    int l=t[o].l,r=t[o].r;
    if(ql<=l&&qr>=r)
     return t[o].sum;
    pushdown(o);
    int mid=(l+r)>>1;
    ll p1=0,p2=0;
    if(ql<=mid) p1=ask((o<<1),ql,qr);
    if(qr>mid) p2=ask((o<<1)|1,ql,qr);
    return (p1+p2);
}
void build(int o,int l,int r)
{
    t[o].l=(ll)l,t[o].r=(ll)r,t[o].siz=(ll)r-l+1;
    if(l==r)
    {
        t[o].sum=val[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson),build(rson);
    update(o);
}
inline ll tree_sum(int u,int v)
{
    ll ans=0;
    while(top[u]!=top[v])
    {
        if(deep[top[u]]<deep[top[v]]) swap(u,v);
        ans=(ans+ask(1,id[top[u]],id[u]));
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) swap(u,v);
    ans=(ans+ask(1,id[v],id[u]));
    return ans;
}
void modify(int o,int ind,ll num)
{
    int l=t[o].l,r=t[o].r;
    if(l==r) 
    {
        t[o].sum+=num;
        return;
    }
    pushdown(o);
    int mid=(l+r)>>1;
    if(ind<=mid) modify((o<<1),ind,num);
    else modify((o<<1)|1,ind,num);
    update(o);
}
int main()
{   
    //freopen("haoi2015_t2.in","r",stdin);
    //freopen("haoi2015_t2.out","w",stdout);
    scanf("%d%d",&n,&m),root=1;
    for(int i=1;i<=n;i++)
     scanf("%lld",&p[i]);
    for(int i=1;i<n;i++)
     {
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
     }
    dfs1(root,0,1);
    dfs2(root,root);
    build(1,1,tot);

    for(int i=1;i<=m;i++)
    {
        int opt,u,v;
        ll x;
        scanf("%d",&opt);
        if(opt==1)
         scanf("%d%lld",&u,&x),adj(1,id[u],id[u],x);
        if(opt==2)
         scanf("%d%lld",&u,&x),adj(1,id[u],id[u]+size[u]-1,x);
        if(opt==3)
         scanf("%d",&u),printf("%lld\n",tree_sum(root,u));
    }

    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值