[SHOI2012]魔法树

2019-07-07 09:04:43 阅读：264 来源： 互联网

标签：node int 魔法 tree tag rc go SHOI2012

省选D2T3考板子可真是不多见呢。。。。~~~

这题就是一个裸的树链剖分，对于每一个Add操作，维护从u至v的路径，对于每一个Query操作，询问以u为根的子树之和。如果不会树链剖分可以看我的往期博客，具体细节在代码之中就不多赘述了~

下面给出参考代码：

#include<iostream>
#include<cstdio>
#define N 400005
#define M 800005
#define lc k*2
#define rc k*2+1
#define mid (l+r)/2
#define int long long
using namespace std;
struct node
{
    int l,r,w,tag;
}tree[4*N];
int n,m,r,p,x,y,z;
int v[M],head[M],nxt[M],cnt;
int dep[N],fa[N],son[N],size[N],top[N],seg[N],id;
char q;
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-48;ch=getchar();}
    return x*f;
}
void add(int a,int b)
{
    v[++cnt]=b;
    nxt[cnt]=head[a];
    head[a]=cnt;
}
void dfs1(int node,int father)
{
    dep[node]=dep[father]+1;
    fa[node]=father;
    size[node]=1;
    int maxson=-1;
    for(int i=head[node];i;i=nxt[i])
    {
        int go=v[i];
        if(go==father)continue;
        dfs1(go,node);
        size[node]+=size[go];
        if(size[go]>maxson)maxson=size[go],son[node]=go;
    }
}
void dfs2(int node,int topfather)
{
    seg[node]=++id;
    top[node]=topfather;
    if(!son[node])return;
    dfs2(son[node],topfather);
    for(int i=head[node];i;i=nxt[i])
    {
        int go=v[i];
        if(go==fa[node]||go==son[node])continue;
        dfs2(go,go);
    }
}
void build(int l,int r,int k)
{
    tree[k].l=l;tree[k].r=r;
    if(l==r)return;
    build(l,mid,lc);
    build(mid+1,r,rc);
}
void pushdown(int k)
{
    tree[lc].tag+=tree[k].tag;
    tree[rc].tag+=tree[k].tag;
    tree[lc].w+=(tree[lc].r-tree[lc].l+1)*tree[k].tag;
    tree[rc].w+=(tree[rc].r-tree[rc].l+1)*tree[k].tag;
    tree[k].tag=0;
    return;
}
void pushup(int k)
{
    tree[k].w=tree[lc].w+tree[rc].w;
    return;
}
void add(int x,int y,int k)
{
    int l=tree[k].l,r=tree[k].r;
    if(l>=x&&r<=y)
    {
        tree[k].tag+=z;
        tree[k].w+=(r-l+1)*z;
        return;
    }
    if(tree[k].tag)pushdown(k);
    if(x<=mid)add(x,y,lc);
    if(y>mid)add(x,y,rc);
    pushup(k);
}
int query(int x,int y,int k)
{
    int l=tree[k].l,r=tree[k].r;
    if(l>=x&&r<=y)
    {
        return tree[k].w;
    }
    if(tree[k].tag)pushdown(k);
    int res=0;
    if(x<=mid)res+=query(x,y,lc);
    if(y>mid)res+=query(x,y,rc);
    return res;
}
void Tadd(int x,int y)
{
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        add(seg[top[x]],seg[x],1);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    add(seg[x],seg[y],1);
}
int Treequery(int x)
{
    return query(seg[x],seg[x]+size[x]-1,1);
}
signed main()
{
    n=read();
    for(int i=1;i<n;i++)
    {
        x=read();y=read();
        x++;y++;
        add(x,y);add(y,x);
    }
    dfs1(1,0);dfs2(1,1);
    build(1,n,1);
    m=read();
    while(m--)
    {
        cin>>q;
        if(q=='A')
        {
            x=read();y=read();z=read();
            x++;y++;
            Tadd(x,y);
        }
        else
        {
            x=read();
            x++;
            cout<<Treequery(x)<<endl;
        }
    }
}

标签：node,int,魔法,tree,tag,rc,go,SHOI2012
来源： https://www.cnblogs.com/szmssf/p/11145035.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

[SHOI2012]魔法树