POJ-2763-Housewife Wind(树剖+LCA,修改边权)

题目链接:http://poj.org/problem?id=2763

题目大意:给出一个树,每条边都有自己的权值,给你一个起点S,一个人从S开始走,有两种状态,1.将第i条边的边权修改为val。0.从当前位置走到T。每次0的时候输出行走的距离。

思路:这个是之前放置有一段时间的题了,当时还不会树剖,没什么思路,最近学过树剖之后,发现可以用树剖来解决。

类似洛谷的一道题:https://blog.csdn.net/qq_40482358/article/details/89853548

同理转化成边权即可。另外还有一个就是由于修改的不是点,而是第几条边,怎么办呢?我的方法是将这些边都存到一个数组中,每次修改的时候查询该边的两个顶点,然后修改即可。会树剖之后就挺简单的(树剖(暴力)大法好)。

ACCode:

//#pragma comment(linker, "/STACK:1024000000,1024000000")
  
#include<stdio.h>
#include<string.h> 
#include<math.h> 
   
#include<map>  
#include<set>
#include<deque> 
#include<queue> 
#include<stack> 
#include<bitset>
#include<string> 
#include<fstream>
#include<iostream> 
#include<algorithm> 
using namespace std; 
  
#define ll long long 
#define Pair pair<int,int>
//#define max(a,b) (a)>(b)?(a):(b)
//#define min(a,b) (a)<(b)?(a):(b)
#define clean(a,b) memset(a,b,sizeof(a))// ??
//std::ios::sync_with_stdio(false);
//  register
const int MAXN=1e5+10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll mod=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
 
class Segment{
	ll Tree[MAXN<<2],Lazy[MAXN<<2];
	
	public:
		void Build(int l,int r,int rt,ll A[]){
			Tree[rt]=Lazy[rt]=0;
			if(l==r){
				Tree[rt]=A[l];
				return ;
			}
			int mid=(l+r)>>1;
			Build(l,mid,rt<<1,A);Build(mid+1,r,rt<<1|1,A);
			Tree[rt]=Tree[rt<<1]+Tree[rt<<1|1];
		}
		void Update(int ql,int qr,int val,int l,int r,int rt){
			if(ql<=l&&r<=qr){
				Tree[rt]=val;
				return ;
			}
			int mid=(l+r)>>1;
			if(ql<=mid) Update(ql,qr,val,l,mid,rt<<1);
			if(qr>mid) Update(ql,qr,val,mid+1,r,rt<<1|1);
			Tree[rt]=Tree[rt<<1]+Tree[rt<<1|1];
		}
		ll Query(int ql,int qr,int l,int r,int rt){
			if(ql<=l&&r<=qr){
				return Tree[rt];
			}
			int mid=(l+r)>>1;
			ll res=0;
			if(ql<=mid) res+=Query(ql,qr,l,mid,rt<<1);
			if(qr>mid) res+=Query(ql,qr,mid+1,r,rt<<1|1);
			return res;
		}
		void Show(int l,int r,int rt){
			printf("l=%d r=%d Tree[rt]=%lld\n",l,r,Tree[rt]);
			if(l==r) return ;
			int mid=(l+r)>>1;
			Show(l,mid,rt<<1);Show(mid+1,r,rt<<1|1);
		}
		
};
struct Node2{
	int u,v;ll val;
	Node2(int _u=0,int _v=0,ll _val=0){
		u=_u;v=_v;val=_val;
	}
};
struct Node1{
	int v,nxt;ll val;
	Node1(int _v=0,ll _val=0,int _nxt=0){
		v=_v;val=_val;nxt=_nxt;
	}
};
Segment Seg;
Node2 EEdge[MAXN<<2];
Node1 Edge[MAXN<<2];
int Head[MAXN],Ecnt;
int Deep[MAXN],Fa[MAXN],Size[MAXN],Son[MAXN];
int Idx[MAXN],Icnt;
int Top[MAXN];
ll A[MAXN],B[MAXN];
int n,m,S,T;
 
void Intt(){
	clean(Head,-1);Ecnt=0;
	clean(Deep,0);clean(Fa,-1);clean(Size,0);clean(Son,-1);
	Icnt=0;
}
void AddEdge(int u,int v,ll val){
	Edge[Ecnt]=Node1(v,val,Head[u]);
	Head[u]=Ecnt++;
}
int DFS1(int u,int fa,int dep,ll val){
	Deep[u]=dep;Fa[u]=fa;Size[u]=1;A[u]=val;
	int maxson=-1;
	for(int i=Head[u];i+1;i=Edge[i].nxt){
		int temp=Edge[i].v;
		if(temp==fa) continue;
		Size[u]+=DFS1(temp,u,dep+1,Edge[i].val);
		if(Size[temp]>maxson){
			maxson=Size[temp];Son[u]=temp;
		}
	}return Size[u];
}
void DFS2(int u,int topfa){
	Idx[u]=++Icnt;Top[u]=topfa;B[Idx[u]]=A[u];
	if(Son[u]==-1) return ;
	DFS2(Son[u],topfa);
	for(int i=Head[u];i+1;i=Edge[i].nxt){
		int temp=Edge[i].v;
		if(Idx[temp]==0){
			DFS2(temp,temp);
		}
	}
}
void UpdatePath(int l,int r,ll val){
//	cout<<"Update:"<<l<<" "<<r<<endl;
	while(Top[l]!=Top[r]){
		if(Deep[Top[l]]<Deep[Top[r]]) swap(l,r);
//		cout<<Idx[Top[l]]<<" "<<Idx[l]<<endl;
		Seg.Update(Idx[Top[l]],Idx[l],val,1,n,1);
		l=Fa[Top[l]];
	}
	if(Deep[l]>Deep[r]) swap(l,r);
//	cout<<Idx[l]+1<<" "<<Idx[r]<<endl;
	Seg.Update(Idx[l]+1,Idx[r],val,1,n,1);
}
ll QueryPath(int l,int r){
	ll ans=0;
	while(Top[l]!=Top[r]){
		if(Deep[Top[l]]<Deep[Top[r]]) swap(l,r);
//		cout<<"Query: "<<Idx[Top[l]]<<" "<<Idx[l]<<"   ans+"<<Seg.Query(Idx[Top[l]],Idx[l],1,n,1)<<endl;
		ans+=Seg.Query(Idx[Top[l]],Idx[l],1,n,1);
		l=Fa[Top[l]];
	}
	if(Deep[l]>Deep[r]) swap(l,r);
//	cout<<"Query: "<<Idx[l]+1<<" "<<Idx[r]<<"   ans+"<<Seg.Query(Idx[l]+1,Idx[r],1,n,1)<<endl;
	ans+=Seg.Query(Idx[l]+1,Idx[r],1,n,1);
	return ans;
}
int main(){
	Intt();
	scanf("%d%d%d",&n,&m,&S);
	for(int i=1;i<n;++i){
		int a,b;ll val;scanf("%d%d%lld",&a,&b,&val);
		EEdge[i]=Node2(a,b,val);
		AddEdge(a,b,val);AddEdge(b,a,val);
	}DFS1(S,-1,1,0);DFS2(S,S);Seg.Build(1,n,1,B);
//	for(int i=1;i<=n;++i){
//		cout<<i<<": "<<Idx[i]<<" "<<A[i]<<" "<<B[Idx[i]]<<endl;
//	}
//	Seg.Show(1,n,1);
	for(int i=1;i<=m;++i){
		int oper;scanf("%d",&oper);
		if(oper){
			int I;ll val;scanf("%d%lld",&I,&val);
			int a=EEdge[I].u,b=EEdge[I].v;
			UpdatePath(a,b,val);
		}
		else{
			scanf("%d",&T);
			printf("%lld\n",QueryPath(S,T));
			S=T;
		}
//		Seg.Show(1,n,1);
	}
	
}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值