POJ2763(树链剖分,边权,模板)

题意:给定一个树形图,某人原来在 s 点,每条边(路)有通过的时间花费,有两种操作:1. 查询某人到 u 点花费的时间 2. 更新某条路的时间花费。

解题思路:前面树链剖分练得一直是点权,但一遇边权懵逼了。下里面的代码是从点权改过来的。其实边权和点权是差不多的。大致意思就是,取每条边连个点中,处于下位的(更深)点来记录这条边的值。有一些查询时的细节和点权稍有不同,详见代码注释。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#pragma comment(linker, "/STACK:102400000,102400000")
#define N 100050
using namespace std;
struct node
{
	int v,next;
}s[2*N];
struct edge
{
	int u,v,c;
}e[N];
struct tree
{
	int l,r,w,c;
}p[3*N];
int w[N],head[N],cnt,fa[N],siz[N],son[N],dep[N],top[N],id[N];
void add_edge(int u,int v)
{
	s[cnt].v=v;
	s[cnt].next=head[u];
	head[u]=cnt++;
	s[cnt].v=u;
	s[cnt].next=head[v];
	head[v]=cnt++;	
}
void dfs1(int u,int f,int deep)
{
	dep[u]=deep;//标记每个点的深度 
    fa[u]=f;//标记每个点的父亲 
    siz[u]=1;//标记每个非叶子节点的子树大小	
    int maxson=-1;
    for(int i=head[u];i!=-1;i=s[i].next)
    {
    	int v=s[i].v;
    	if(v==f) continue;
    	dfs1(v,u,deep+1);
    	siz[u]+=siz[v];
    	if(siz[v]>maxson)
    	{
    		son[u]=v;
    		maxson=siz[v];
		}
	}
}
void dfs2(int u,int t)
{
	id[u]=++cnt;
	top[u]=t;
	if(son[u]==-1) return;
	dfs2(son[u],t);
	for(int i=head[u];i!=-1;i=s[i].next)
	{
		int v=s[i].v;
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	}
}
void build(int rt,int l,int r)
{
	p[rt].l=l,p[rt].r=r,p[rt].c=0,p[rt].w=0;
	if(l==r) return;
	int mid=(l+r)/2;
	build(2*rt,l,mid);
	build(2*rt+1,mid+1,r);
}
int query(int rt,int l,int r)
{
	if(l==p[rt].l&&r==p[rt].r)
	{
		return p[rt].w;
	}
	int mid=(p[rt].l+p[rt].r)/2;
	if(r<=mid)
	{
		return query(2*rt,l,r);
	}else
	if(l>=mid+1)
	{
		return query(2*rt+1,l,r);
	}else
	{
		return (query(2*rt+1,mid+1,r)+query(2*rt,l,mid));
	}
}
void update(int rt,int l,int r,int kk)
{
	if(l==p[rt].l&&r==p[rt].r)
	{
		p[rt].w=kk;
		return;
	}
	int mid=(p[rt].l+p[rt].r)/2;
	if(r<=mid)
	{
		update(2*rt,l,r,kk);
	}else
	if(l>=mid+1)
	{
		update(2*rt+1,l,r,kk);
	}else
	{
		update(2*rt+1,mid+1,r,kk);
		update(2*rt,l,mid,kk);
	}	
	p[rt].w=(p[2*rt].w+p[2*rt+1].w);
}
long long qrange(int u,int v)
{
	long long res=0;
	while(top[u]!=top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		res+=query(1,id[top[u]],id[u]);
		u=fa[top[u]];
	}
	if(u==v) return res;//必须加上这句话,因为下面要用到son[u],万一u==v的话就永远也找不到,会runtime error 
	if(dep[u]>dep[v]) swap(u,v);
	res+=query(1,id[son[u]],id[v]);//最后汇合点的上位边是不能算入的,这点一定要注意 
	return res;
}
void uprange(int u,int v,int k)
{
	while(top[u]!=top[v])
	{
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		update(1,id[top[u]],id[u],k);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	update(1,id[u],id[v],k);
}
int main()
{
	//freopen("t.txt","r",stdin);
	int n,m,a,b,q,tp,st;
	scanf("%d%d%d",&n,&q,&st);
	cnt=0;
	memset(fa,-1,sizeof(fa));
	memset(head,-1,sizeof(head));
	memset(son,-1,sizeof(son));
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].c);
		add_edge(e[i].u,e[i].v); 
	}
	dfs1(1,-1,1);
	cnt=0;
	dfs2(1,1);
	build(1,1,n);
	for(int i=1;i<n;i++) 
		if(dep[e[i].u]>dep[e[i].v]) uprange(e[i].u,e[i].u,e[i].c);
		else uprange(e[i].v,e[i].v,e[i].c);
	while(q--)
	{
		scanf("%d",&tp);
		if(tp==1)
		{
			scanf("%d%d",&a,&b);
			if(dep[e[a].u]>dep[e[a].v]) uprange(e[a].u,e[a].u,b);
				else uprange(e[a].v,e[a].v,b);
		}else
		{
			scanf("%d",&a);
			printf("%d\n",qrange(st,a));
			st=a;
		}
	}
	return 0;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值