hdu3966_树链剖分

最近在强化知识点深度,发现树链剖分不是很会写了。

回顾一下修改操作:

若两个点在同一条链上,则直接修改这段区间。

若不在同一条链上,修改深度较大的点到其链顶端的区间,同时将这个点变为他所在链顶端的父亲,循环操作直到这两个点在同一条链上,就可以用上一种方法了。

没有用LCA写是因为以前被坑过,不但没有这种方法好写,效率也不太让人满意。

主要是对第二种情况如何写有所遗忘,写道模版再给自己提个醒。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define MAXN 50005
int n,m,q;
vector<int> map[MAXN];
int size[MAXN],fa[MAXN],son[MAXN],val[MAXN],tid[MAXN],_tid[MAXN],dep[MAXN],top[MAXN];
int cnt;
struct node
{
    int l,r,val;
}tree[MAXN<<2];
void dfs1(int s,int f,int d)
{
    size[s]=1;
    fa[s]=f;
    dep[s]=d;
    int len=map[s].size();
    for(int i=0;i<len;i++)
    {
        int e=map[s][i];
        if(e==f)continue;
        dfs1(e,s,d+1);
        size[s]+=size[e];
        if(son[s]==0)
        son[s]=e;
        else if(size[son[s]]<size[e])
        son[s]=e;
    }
}
void dfs2(int s,int t)
{
    tid[s]=++cnt;
    _tid[cnt]=s;
    top[s]=t;
    if(son[s]!=0)
    dfs2(son[s],t);
    int len=map[s].size();
    for(int i=0;i<len;i++)
    {
        int e=map[s][i];
        if(e!=fa[s] && e!=son[s])
        dfs2(e,e);
    }
}
void build(int l,int r,int now)
{
    tree[now].l=l;
    tree[now].r=r;
    tree[now].val=0;
    if(l==r)
    {
        tree[now].val=val[_tid[l]];
        return ;
    }
    int mid=(l+r)>>1;
    build(l,mid,now<<1);
    build(mid+1,r,now<<1|1);
}
void down(int now)
{
    tree[now<<1].val+=tree[now].val;
    tree[now<<1|1].val+=tree[now].val;
    tree[now].val=0;
}
void update(int l,int r,int now,int num)
{
    if(l==tree[now].l && r==tree[now].r)
    {
        tree[now].val+=num;
        return ;
    }
    if(tree[now].val)
    down(now);
    int mid=(tree[now].l+tree[now].r)>>1;
    if(r<=mid)
    update(l,r,now<<1,num);
    else if(l>mid)
    update(l,r,now<<1|1,num);
    else
    {
        update(l,mid,now<<1,num);
        update(mid+1,r,now<<1|1,num);
    }
}
void change(int s,int e,int num)
{
    while(top[s]!=top[e])
    {
        if(dep[top[s]]<dep[top[e]])
        swap(s,e);
        update(tid[top[s]],tid[s],1,num);
        s=fa[top[s]];
    }
    if(dep[s]>dep[e])
    swap(s,e);
    update(tid[s],tid[e],1,num);
}
int query(int l,int now)
{
    if(tree[now].l==l && tree[now].r==l)
    return tree[now].val;
    if(tree[now].val)
    down(now);
    int mid=(tree[now].l+tree[now].r)>>1;
    if(l<=mid)
    return query(l,now<<1);
    else
    return query(l,now<<1|1);
}
int main()
{
    while(cin>>n>>m>>q)
    {
        memset(size,0,sizeof(size));
        memset(fa,0,sizeof(fa));
        memset(tid,0,sizeof(tid));
        memset(son,0,sizeof(son));
        cnt=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&val[i]);
            map[i].clear();
        }
        for(int i=1;i<n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            map[a].push_back(b);
            map[b].push_back(a);
        }
        dfs1(1,-1,1);
        dfs2(1,1);
        build(1,n,1);
        while(q--)
        {
            char ch[5];
            int a,b,c;
            scanf("%s",ch);
            if(ch[0]=='Q')
            {
                scanf("%d",&a);
                printf("%d\n",query(tid[a],1));
            }
            else
            {
                scanf("%d%d%d",&a,&b,&c);
                if(ch[0]=='I')
                change(a,b,c);
                else
                change(a,b,-c);
            }
        }
    }
}
时间: 2024-11-05 19:03:48

hdu3966_树链剖分的相关文章

BZOJ 2243: [SDOI2011]染色 树链剖分

2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 1886  Solved: 752[Submit][Status] Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“112221”由3段组成:“11”.“222”和“1”. 请你写一个程序依次完成这m个操作. In

bzoj 2243: [SDOI2011]染色 线段树区间合并+树链剖分

2243: [SDOI2011]染色 Time Limit: 20 Sec  Memory Limit: 512 MBSubmit: 7925  Solved: 2975[Submit][Status][Discuss] Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“112221”由3段组成:“11”.“222”和“1”. 请你写一个程序依次完

bzoj3694: 最短路(树链剖分/并查集)

bzoj1576的帮我们跑好最短路版本23333(双倍经验!嘿嘿嘿 这题可以用树链剖分或并查集写.树链剖分非常显然,并查集的写法比较妙,涨了个姿势,原来并查集的路径压缩还能这么用... 首先对于不在最短路径树上的边x->y,设t为最短路径树上lca(x,y),则t到y上的路径上的点i到根的距离都可以用h[x]+dis[x][y]+h[y]-h[i](h[]为深度)来更新,因为h[i]一定,只要让h[x]+dis[x][y]+h[y]最小就行,这里用树剖直接修改整条链上的数,就可以过了. 并查集的

洛谷 P3384 【模板】树链剖分

题目描述 如题,已知一棵包含N个结点的树(连通且无环),每个节点上包含一个数值,需要支持以下操作: 操作1: 格式: 1 x y z 表示将树从x到y结点最短路径上所有节点的值都加上z 操作2: 格式: 2 x y 表示求树从x到y结点最短路径上所有节点的值之和 操作3: 格式: 3 x z 表示将以x为根节点的子树内所有节点值都加上z 操作4: 格式: 4 x 表示求以x为根节点的子树内所有节点值之和 输入输出格式 输入格式: 第一行包含4个正整数N.M.R.P,分别表示树的结点个数.操作个数

bzoj1036 树的统计(树链剖分+线段树)

1036: [ZJOI2008]树的统计Count Time Limit: 10 Sec  Memory Limit: 162 MBSubmit: 15120  Solved: 6141[Submit][Status][Discuss] Description 一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w.我们将以下面的形式来要求你对这棵树完成一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I

SPOJ QTREE Query on a tree ——树链剖分 线段树

[题目分析] 垃圾vjudge又挂了. 树链剖分裸题. 垃圾spoj,交了好几次,基本没改动却过了. [代码](自带常数,是别人的2倍左右) #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; #define maxn 20005 int T,n,fr[maxn],h[maxn],to[maxn],ne[maxn]

树链剖分简(单)介(绍)

树链剖分可以算是一种数据结构(一大堆数组,按照这个意思,主席树就是一大堆线段树).将一棵树分割成许多条连续的树链,方便完成一下问题: 单点修改(dfs序可以完成) 求LCA(各种乱搞也可以) 树链修改(修改任意树上两点之间的唯一路径) 树链查询 (各种操作)  前两个内容可以用其他方式解决,但是下面两种操作倍增.st表,dfs序就很难解决(解决当然可以解决,只是耗时长点而已).下面开始步入正题. 树链剖分的主要目的是分割树,使它成一条链,然后交给其他数据结构(如线段树,Splay)来进行维护.常

bzoj1146整体二分+树链剖分+树状数组

其实也没啥好说的 用树状数组可以O(logn)的查询 套一层整体二分就可以做到O(nlngn) 最后用树链剖分让序列上树 1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #include<algorithm> 5 using namespace std; 6 inline int read() 7 { 8 int x=0,f=1,ch=getchar(); 9 while(ch<

BZOJ 1036: [ZJOI2008]树的统计Count [树链剖分]

1036: [ZJOI2008]树的统计Count Time Limit: 10 Sec  Memory Limit: 162 MBSubmit: 14302  Solved: 5779[Submit][Status][Discuss] Description 一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w.我们将以下面的形式来要求你对这棵树完成一些操作: I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I