[牛客2020第三场]牛半仙的妹儿Tree

牛半仙的妹儿Tree

题解

其实是很板子的一道树剖，但Indra大佬用分块做出来了

我们发现，如果我们将3操作与之前的1操作相匹配，可以发现，3操作的答案为。

上面这个式子可以拆分为。

只有当上面式子的值应该是小于时它才可以不无视牛半仙。

我们假设使得上式最小，故。

可得。

由于从到上的每一个点都有可能成为之后的祖先，而对于点最小的一定会在上取得。

我们可以先将每个点的初值赋值为，之后每次1操作对到加上的标记，查询就查询到上是否有值是小于的，这些操作都可以用树链剖分来进行维护。

时间复杂度。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define MAXN 100005
typedef long long LL;
const int INF=0x3f3f3f3f;
typedef unsigned long long uLL;
typedef pair<int,int> pii;
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
int n,m,head[MAXN],tot,tr[MAXN<<2],lzy[MAXN<<2],sum[MAXN<<2];
int father[MAXN],dep[MAXN],dfn[MAXN],idx,wson[MAXN],siz[MAXN];
int pre[MAXN],ltp[MAXN];
bool clen[MAXN<<2];
struct edge{int to,nxt;}e[MAXN<<1];
void addEdge(int u,int v){e[++tot]=(edge){v,head[u]};head[u]=tot;}
void dfs1(int u,int fa){
	dep[u]=dep[fa]+1;father[u]=fa;siz[u]=1;
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;
		dfs1(v,u);siz[u]+=siz[v];
		if(siz[wson[u]]<siz[v])wson[u]=v;
	}
}
void dfs2(int u,int tp){
	dfn[u]=++idx;pre[idx]=u;ltp[u]=tp;//printf("dfn%d:%d\n",u,dfn[u]);
	if(!wson[u])return ;dfs2(wson[u],tp);
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v==father[u]||v==wson[u])continue;
		dfs2(v,v);
	}
}
void build(int rt,int l,int r){
	lzy[rt]=tr[rt]=INF;if(l==r)return (void)(sum[rt]=2*dep[pre[l]]);
	int mid=l+r>>1;build(rt<<1,l,mid);build(rt<<1|1,mid+1,r);
	sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
}
void cleanup(int rt){tr[rt]=INF;lzy[rt]=INF;clen[rt]=1;}
void pushdown(int rt){
	if(clen[rt]){clen[rt]=0;cleanup(rt<<1);cleanup(rt<<1|1);}
	if(lzy[rt]<INF-1){
		lzy[rt<<1]=min(lzy[rt<<1],lzy[rt]);
		lzy[rt<<1|1]=min(lzy[rt<<1|1],lzy[rt]);
		tr[rt<<1]=min(lzy[rt<<1]-sum[rt<<1],tr[rt<<1]);
		tr[rt<<1|1]=min(lzy[rt<<1|1]-sum[rt<<1|1],tr[rt<<1|1]);
	}
}
void insert(int rt,int l,int r,int al,int ar,int aw){
	if(al>r||ar<l||l>r)return ;
	if(al<=l&&r<=ar){lzy[rt]=min(lzy[rt],aw);tr[rt]=min(tr[rt],aw-sum[rt]);return ;}
	int mid=l+r>>1;pushdown(rt);
	if(al<=mid)insert(rt<<1,l,mid,al,ar,aw);
	if(ar>mid)insert(rt<<1|1,mid+1,r,al,ar,aw);
	tr[rt]=min(lzy[rt]-sum[rt],min(tr[rt<<1],tr[rt<<1|1]));
	//printf("%d %d %d %d\n",rt,l,r,tr[rt]);
}
void Chain_Insert(int u,int v,int w){
	for(int fu=ltp[u],fv=ltp[v];fu^fv;u=father[fu],fu=ltp[u]){
		if(dep[fu]<dep[fv])swap(u,v),swap(fu,fv);
		insert(1,1,n,dfn[fu],dfn[u],w);
	}
	if(dep[u]>dep[v])swap(u,v);
	insert(1,1,n,dfn[u],dfn[v],w);
}
int query(int rt,int l,int r,int al,int ar){
	if(l>ar||r<al||l>r)return INF;if(al<=l&&r<=ar)return tr[rt];
	int res=INF,mid=l+r>>1;pushdown(rt);
	if(al<=mid)res=min(res,query(rt<<1,l,mid,al,ar));
	if(ar>mid)res=min(res,query(rt<<1|1,mid+1,r,al,ar));
	return res;
}
int Chain_Query(int u,int v){
	int res=INF;
	for(int fu=ltp[u],fv=ltp[v];fu^fv;u=father[fu],fu=ltp[u]){
		if(dep[fu]<dep[fv])swap(u,v),swap(fu,fv);
		res=min(res,query(1,1,n,dfn[fu],dfn[u]));
	}
	if(dep[u]>dep[v])swap(u,v);
	res=min(res,query(1,1,n,dfn[u],dfn[v]));
	return res;
}
signed main(){
	read(n);read(m);
	for(int i=1;i<n;i++){
		int u,v;read(u);read(v);
		addEdge(u,v);addEdge(v,u);
	}
	dfs1(1,0);dfs2(1,0);build(1,1,n);
	for(int i=1;i<=m;i++){
		int opt,x;read(opt);read(x);
		if(opt==1)Chain_Insert(x,1,dep[x]+i);
		if(opt==2)cleanup(1);
		if(opt==3)puts(Chain_Query(x,1)<=i-dep[x]?"wrxcsd":"orzFsYo");
	}
	return 0;
}

谢谢！！！