HDU 3966（树链剖分+点修改+点查询）

[b]题目链接：[/b]http://acm.hdu.edu.cn/showproblem.php?pid=3966

题目大意：营地的分布成树型。每个营地都有一些人，每次修改修改一条链上的所有营地的人数，每次查询单个点。

解题思路：

树链剖分基础题。

维护一个sum。

注意轻链修改时，点修改和边修改的不同。

由于树的结构与线段树点的顺序不太相同，因此需要做一个映射数组rank。故在线段树Build的时候，权值是camp[rank[l]]，rank这步的映射在dfs2的时候完成，rank[w[u]]=u;

Query单点u的时候, w[u]是其在线段树中的位置。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include "cstdio"
#include "vector"
#include "cstring"
using namespace std;
#define maxn 50005
int camp[maxn],s[maxn],dep[maxn],pos[maxn],son[maxn],top[maxn],fa[maxn],w[maxn],rank[maxn],cnt,n;
vector<int> G[maxn];
void dfs1(int u,int pre,int d)
{
s[u]=1;fa[u]=pre;dep[u]=d;
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v==pre) continue;
dfs1(v,u,d+1);
s[u]+=s[v];
if(son[u]!=-1||s[v]>s[son[u]]) son[u]=v;
}
}
void dfs2(int u,int tp)
{
w[u]=++cnt;top[u]=tp;rank[w[u]]=u;
if(son[u]==-1) return;
dfs2(son[u],tp);
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i];
if(v!=son[u]&&v!=fa[u])
dfs2(v,v);
}
}
//Segment-Tree
#define lson l,mid,root<<1
#define rson mid+1,r,root<<1|1
int col[maxn<<2],sum[maxn<<2];
void PushUp(int root)
{
sum[root]=sum[root<<1]+sum[root<<1|1];
}
void Build(int l,int r,int root)
{
col[root]=0;
if(l==r)
{
sum[root]=camp[rank[l]];
return;
}
int mid=(l+r)>>1;
Build(lson);
Build(rson);
PushUp(root);
}
void PushDown(int root,int m)
{
if(col[root])
{
col[root<<1]+=col[root];
col[root<<1|1]+=col[root];
sum[root<<1]+=(m-(m>>1))*col[root];
sum[root<<1|1]+=(m>>1)*col[root];
col[root]=0;
}
}
void Update(int L,int R,int v,int l,int r,int root)
{
if(L<=l&&r<=R)
{
col[root]+=v;
sum[root]+=(r-l+1)*v;
return;
}
PushDown(root,r-l+1);
int mid=(l+r)>>1;
if(L<=mid) Update(L,R,v,lson);
if(R>mid) Update(L,R,v,rson);
PushUp(root);
}
int Query(int p,int l,int r,int root)
{
if(l==r) return sum[root];
PushDown(root,r-l+1);
int mid=(l+r)>>1,ret=0;
if(p<=mid) ret=Query(p,lson);
else ret=Query(p,rson);
return ret;
}
void Change(int x,int y,int v)
{
while(top[x]!=top[y])
{
if(dep[top[x]]<dep[top[y]]) swap(x,y);
Update(w[top[x]],w[x],v,1,n,1);
x=fa[top[x]];
}
if(dep[x]>dep[y]) swap(x,y);
Update(w[x],w[y],v,1,n,1);//与边修改不同
}
int main()
{
//freopen("in.txt","r",stdin);
int m,p,u,v,c;
while(scanf("%d%d%d",&n,&m,&p)!=EOF)
{
memset(son,-1,sizeof(son));
for(int i=0;i<=n;i++) G[i].clear();
cnt=0;
for(int i=1;i<=n;i++)
scanf("%d",&camp[i]);
for(int i=1;i<=m;i++)
{
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs1(1,1,1);
dfs2(1,1);
Build(1,n,1);
char cmd[10];
for(int i=1;i<=p;i++)
{
scanf("%s",&cmd);
if(cmd[0]=='I')
{
scanf("%d%d%d",&u,&v,&c);
Change(u,v,c);
}
if(cmd[0]=='D')
{
scanf("%d%d%d",&u,&v,&c);
Change(u,v,-c);
}
if(cmd[0]=='Q')
{
scanf("%d",&c);
printf("%d\n",Query(w[c],1,n,1));
}
}
}

}

11757337

2014-09-29 16:51:29

Accepted

3966

2484MS

8308K

3299 B

C++

Physcal