HDU 3966

树链剖分 练模板;

用的 是HH的线段树 虽然之前是我不用的摸板

修改区间 求点值;

CODE:

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <stack>
#include <vector>
#include <set>
#include <map>
using namespace std;
#define N 503456

struct edge
{
int v,next;
}e[N<<1];
int head[N<<1],tot;
int top[N];
int fa[N],dep[N],sz[N],son[N],p[N],fp[N],pos,a[N];

void add(int u,int v)
{
e[tot].v=v;
e[tot].next=head[u];
head[u]=tot++;
}

void init()
{
tot=pos=0;
memset(head,-1,sizeof(head));
memset(son,-1,sizeof(son));
}

void dfs(int u,int pre,int d)
{
dep[u]=d;
sz[u]=1;
fa[u]=pre;
for (int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].v;
if (v==pre) continue;
dfs(v,u,d+1);
sz[u]+=sz[v];
if (son[u]==-1||sz[son[u]]<sz[v]) son[u]=v;
}
}

void getpos(int u,int sp)
{
top[u]=sp;
p[u]=++pos;
fp[pos]=u;
if (son[u]==-1) return;
getpos(son[u],sp);
for (int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].v;
if (v!=son[u]&&v!=fa[u])
getpos(v,v);
}
}

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[N<<2],col[N<<2];

void pushup(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int m)
{
if (col[rt])
{
col[rt<<1]+=col[rt];
col[rt<<1|1]+=col[rt];
sum[rt<<1]+=col[rt]*(m-(m>>1));
sum[rt<<1|1]+=col[rt]*(m>>1);
col[rt]=0;
}
}
void build(int l,int r,int rt)
{
col[rt]=0;
if (l==r)
{
sum[rt]=a[fp[l]];
return;
}
int m=(l+r)>>1;
build(lson);
build(rson);
pushup(rt);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if (L<=l&&r<=R)
{
col[rt]+=c;
sum[rt]+=c*(r-l+1);
return;
}
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if (L<=m) update(L,R,c,lson);
if (m<R) update(L,R,c,rson);
pushup(rt);
}

int query(int p,int l,int r,int rt)
{
if (l==r) return sum[rt];
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if (p<=m) return query(p,lson);
if (p>m) return query(p,rson);
}

void lca(int u,int v,int c)
{
int fu=top[u],fv=top[v];
while (fu!=fv)
{
if (dep[fu]<dep[fv])
{
swap(fu,fv);
swap(u,v);
}
update(p[fu],p[u],c,1,pos,1);
u=fa[fu];
fu=top[u];
}
if (dep[u]>dep[v]) swap(u,v);
update(p[u],p[v],c,1,pos,1);
}

int main()
{
int n,m,t,u,v,w;
while (scanf("%d%d%d",&n,&m,&t)!=EOF)
{
init();
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
for (int i=1;i<=m;i++)
{
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
dfs(1,0,0);
getpos(1,1);
build(1,pos,1);
char op[10];
while (t--)
{
scanf("%s",op);
if (op[0]==‘Q‘)
{
scanf("%d",&u);
printf("%d\n",query(p[u],1,pos,1));
}
else
{
scanf("%d%d%d",&u,&v,&w);
if (op[0]==‘D‘) w=-w;
lca(u,v,w);
}
}
}
return 0;
}

时间: 2024-12-24 01:21:17

HDU 3966的相关文章

HDU 3966 Aragorn&#39;s Story

题目:Aragorn's Story 链接:http://acm.hdu.edu.cn/showproblem.php?pid=3966 题意:给一棵树,每个结点都有初始的权值,有m个操作,分两种:一是从x 结点到y 结点路上所有的结点权值+z或-z,二是问x结点的权值. 思路: 树链剖分. 这是我学树剖的第一题,建议还没接触过的伙伴,第一次学习的时候不要一直纠结理论,直接找一道模板题,然后找一篇AC代码,直接理解,做完一题后,你就会发现理论其实也挺好理解的,树剖也挺好学的. 树剖中的新概念:重

HDU 3966 Aragorn&#39;s Story (树链点权剖分,成段修改单点查询)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3966 树链剖分的模版,成段更新单点查询.熟悉线段树的成段更新的话就小case啦. 1 //树链剖分 边权修改 单点查询 2 #include <iostream> 3 #include <cstring> 4 #include <algorithm> 5 #include <cstdio> 6 using namespace std; 7 const int M

Hdu 3966 Aragorn&#39;s Story (树链剖分 + 线段树区间更新)

题目链接: Hdu 3966 Aragorn's Story 题目描述: 给出一个树,每个节点都有一个权值,有三种操作: 1:( I, i, j, x ) 从i到j的路径上经过的节点全部都加上x: 2:( D, i, j, x ) 从i到j的路径上经过的节点全部都减去x: 3:(Q, x) 查询节点x的权值为多少? 解题思路: 可以用树链剖分对节点进行hash,然后用线段树维护(修改,查询),数据范围比较大,要对线段树进行区间更新 1 #include <cstdio> 2 #include

AC日记——Aragorn&#39;s Story HDU 3966

Aragorn's Story Time Limit: 10000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 10510    Accepted Submission(s): 2766 Problem Description Our protagonist is the handsome human prince Aragorn comes from The Lor

(第一个树链剖分!!!!)hdu 3966

A - Aragorn's Story Time Limit:3000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u Submit Status Practice HDU 3966 Description Our protagonist is the handsome human prince Aragorn comes from The Lord of the Rings. One day Aragorn finds

HDU 3966 Aragorn&#39;s Story(树链剖分 模板题)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3966 Problem Description Our protagonist is the handsome human prince Aragorn comes from The Lord of the Rings. One day Aragorn finds a lot of enemies who want to invade his kingdom. As Aragorn knows, th

树链剖分 [HDU 3966] Aragorn&#39;s Story

Aragorn's Story Time Limit: 10000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others) Total Submission(s): 3544    Accepted Submission(s): 995 Problem Description Our protagonist is the handsome human prince Aragorn comes from The Lord

HDU 3966 Aragorn&amp;#39;s Story(树链剖分)

HDU Aragorn's Story 题目链接 树抛入门裸题,这题是区间改动单点查询,于是套树状数组就OK了 代码: #include <cstdio> #include <cstring> #include <vector> #include <algorithm> using namespace std; const int N = 50005; inline int lowbit(int x) {return x&(-x);} int dep

HDU 3966 Aragorn&#39;s Story 树链剖分

Link: http://acm.hdu.edu.cn/showproblem.php?pid=3966 这题注意要手动扩栈. 这题我交g++无限RE,即使手动扩栈了,但交C++就过了. 1 #pragma comment(linker, "/STACK:1024000000,1024000000") 2 #include <cstdio> 3 #include <cstring> 4 #include <iostream> 5 #include &

HDU 3966 (树链剖分+线段树)

Problem Aragorn's Story (HDU 3966) 题目大意 给定一颗树,有点权. 要求支持两种操作,将一条路径上的所有点权值增加或减少ai,询问某点的权值. 解题分析 树链剖分模板题. 实质上树链剖分进行了点对点的一次映射,保证了重链上的点在线段树上的位置是连续的. 树链剖分的两个性质(转): 性质1:如果(v,u)为轻边,则siz[u] * 2 < siz[v]: 性质2:从根到某一点的路径上轻链.重链的个数都不大于logn. 保证了一个区间的时间复杂度是log2(n).