SNOI省選模擬賽Round1 T3 tree

題目大意:

給定一顆樹,要求支持兩種操作:

1 x y:加入一組點對,並輸出x到y路徑上的和。

2 e v:將第e條邊的值改爲v,然後詢問之前加入的點對距離和的最大值,然後恢復原狀。

題解:樹鏈剖分。

我太弱了並不能完全講清題解的意思,就先佔個坑吧。

代碼:

#include<bits/stdc++.h>
#define maxn 300005
#define ls (id<<1)
#define rs (id<<1|1)
#define fir maxa[id].first
#define sec maxa[id].second
#define pp pair<int,int>
using namespace std;
typedef long long LL;
int read()
{
	char c;int sum=0,f=1;c=getchar();
	while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
	while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}
	return sum*f;
}
int head[maxn],to[maxn<<1],nex[maxn<<1],cnt,val[maxn<<1];
void addedge(int u,int v,int w)
{
	to[++cnt]=v;nex[cnt]=head[u],val[cnt]=w;head[u]=cnt;
}
int n,m,w[maxn],pos[maxn];
void input()
{
	n=read();m=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read();w[i]=read();
		addedge(u,v,w[i]),addedge(v,u,w[i]);
	}
}
struct node{
	pp maxa[maxn<<2];
	void modify(int id,int l,int r,int ql,int qr,int v,int opt)
	{
		if(ql==l && r==qr)
		{
			if(opt==1) sec=max(sec,v);
			else fir=max(fir,v);
			return;
		}
		int mid=l+r>>1;
		if(qr<=mid)modify(ls,l,mid,ql,qr,v,opt);
		else if(ql>mid)modify(rs,mid+1,r,ql,qr,v,opt);
		else modify(ls,l,mid,ql,mid,v,opt),modify(rs,mid+1,r,mid+1,qr,v,opt);
	}
	pp query(int id,int l,int r,int pos)
	{
		if(l==r)
		return maxa[id];
		int mid=(l+r)>>1;
		pp ret; 
		if(pos<=mid) ret=query(ls,l,mid,pos);
		else ret=query(rs,mid+1,r,pos);
		ret.first=max(ret.first,maxa[id].first);
		ret.second=max(ret.second,maxa[id].second);
		return ret;
	}
}tree;
int fa[maxn],deep[maxn],dis[maxn],in[maxn],top[maxn],out[maxn],cur,anc[maxn][20],siz[maxn];//,pw[maxn];
void dfs1(int x,int la)
{
	siz[x]=1;
	fa[x]=la;
	deep[x]=deep[la]+1;
	anc[x][0]=la;
	for(int i=1;i<20;++i)
	anc[x][i]=anc[anc[x][i-1]][i-1];
	for(int i=head[x];i;i=nex[i])
	{
		int v=to[i];
		if(v==la)
		continue;
		dis[v]=dis[x]+val[i];
		pos[i>>1]=v;
		dfs1(v,x);
		siz[x]+=siz[v];
	}
}
void dfs2(int x,int la,int tp)
{
	in[x]=++cur;
	top[x]=tp;
	int son=0;
	siz[son]=0;
	for(int i=head[x];i;i=nex[i])
	{
		int v=to[i];
		if(v==la) continue;
		if(siz[v]>siz[son])
		son=v;
	}
	if(son)
	dfs2(son,x,tp);
	for(int i=head[x];i;i=nex[i])
	{
		int v=to[i];
		if (v==la || v==son) continue;
		dfs2(v,x,v);
	}
	out[x]=cur;
}
int LCA(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]]) swap(x,y);
		x=fa[top[x]];
	}
	if(deep[x]>deep[y]) swap(x,y);
	return x;
}
void add(int x,int p,int v)
{
	while(top[x]!=top[p])
	{
		tree.modify(1,1,n,in[top[x]],in[x],v,0);
		if(out[x]!=out[top[x]])
		tree.modify(1,1,n,out[x]+1,out[top[x]],v,1);
		int t=fa[top[x]];
		if(in[t]+1<in[top[x]])
		tree.modify(1,1,n,in[t]+1,in[top[x]]-1,v,1);
		if(out[x]!=out[t])
		tree.modify(1,1,n,out[x]+1,out[t],v,1);
		x=t;
	}
	tree.modify(1,1,n,in[p],in[x],v,0);
	if(out[x]!=out[p])
	tree.modify(1,1,n,out[x]+1,out[p],v,1);
}
int Jump(int x,int dis)
{
	for(int i=0;i<20 && dis;++i,dis>>=1)
	if(dis&1)
	x=anc[x][i];
	return x;
}
void solve_add(int x,int y)
{
	if(x==y) return;
	if(deep[x]>deep[y]) swap(x,y);
	int lca=LCA(x,y),ans=dis[x]+dis[y]-(dis[lca]*2);
	printf("%d\n",ans);
	int px=Jump(x,deep[x]-deep[lca]-1),py=Jump(y,deep[y]-deep[lca]-1);
	if(x!=lca)
	add(x,px,ans);
	add(y,py,ans);
	if(in[lca]>1) tree.modify(1,1,n,1,in[lca]-1,ans,1);
	if(out[lca]<n) tree.modify(1,1,n,out[lca]+1,n,ans,1);
	if(x!=lca && in[x]!=out[x]) tree.modify(1,1,n,in[x]+1,out[x],ans,1);
	if(in[y]!=out[y]) tree.modify(1,1,n,in[y]+1,out[y],ans,1);
	if(in[px]>in[py]) swap(px,py);
	if(x!=lca)
	{
		if(in[px]>in[lca]) tree.modify(1,1,n,in[lca],in[px]-1,ans,1);
		if(out[px]+1<in[py]) tree.modify(1,1,n,out[px]+1,in[py]-1,ans,1);
		if(out[py]<out[lca]) tree.modify(1,1,n,out[py]+1,out[lca],ans,1);
	}
	else
	{
		if(in[lca]<in[py]) tree.modify(1,1,n,in[lca],in[py]-1,ans,1);
		if(out[py]<out[lca]) tree.modify(1,1,n,out[py]+1,out[lca],ans,1);
	}
}
void solve_ask(int id,int v)
{
	pp p=tree.query(1,1,n,in[pos[id]]);
	if(p.first && p.second)
	printf("%d\n",max(p.first-w[id]+v,p.second));
	else if(!p.first)
	printf("%d\n",p.second);
	else printf("%d\n",p.first-w[id]+v);
}
void solve()
{
	dfs1(1,0);
	dfs2(1,0,1);
	int opt,x,y;
	while(m--)
	{
		opt=read(),x=read(),y=read();
		if(opt==1)
		solve_add(x,y);
		else solve_ask(x,y);
	}
}
int main()
{
	input();
	solve();
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章