342E Xenia and Tree——codeforces

第一次在cf上做题。。外国网站好高端的啊。。


这个一定是有优化的   如果说到优化    第一反应预处理
 但是又开不下一个100000^2的数组  绝对mle


神犇以前给我讲了分块大法   
现在就有用了   分块 我认为就是一个  经过优化的暴力  一般将数据分成sqrt(n)部分  对于每一部分进行一次处理
将处理的值留下来  作为一段操作的预处理   来进行下一段操作   
而对于100000这个数据   分块标准100算是最优了

于是就走着
开一个dis[]表示i号点到最近的红点的距离
直到有100个未处理的红点操作
就统一bfs(时间复杂度o(n))

更新一部分dis

而对于每一个求值操作
就在当前有限的100以内红点(设数量m)
跑m次lca
在与dis数组已处理了的  取一个min
就作为ans输出


#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
using namespace std;
queue<int>q;
int cs;
int n,m;
struct EDGE
{
	int u,v,next;
}e[200000+10];
int cnt;int head[100000+10];
int fa[100000+10][19];
int red[100+10];int rnt;
int depth[100000+10];
int dis[100000+10];
int ans;
void adde(int u,int v)
{
	e[++cnt].u=u;
	e[cnt].v=v;
	e[cnt].next=head[u];	
	head[u]=cnt;
}
void readdata()
{
	scanf("%d%d",&n,&m);
	int u,v;
	for(int i=1;i<=n-1;i++)
	{
		scanf("%d%d",&u,&v);
		adde(u,v);
		adde(v,u);
	}	
	e[0].next=0;
}
void dfs(int u)
{
	for(int i=head[u];e[i].u==u;i=e[i].next)
	{
		if(depth[e[i].v]==0)
		{
			depth[e[i].v]=depth[u]+1;
			fa[e[i].v][0]=u;
			for(int j=1;1<<j<=depth[e[i].v];j++)
			fa[e[i].v][j]=fa[fa[e[i].v][j-1]][j-1];
			dfs(e[i].v);
		}
	}
}
int go_up(int uu,int delta)
{
	for(int i=17;i>=0;i--)
	{
		if(delta==0)
		break;
		if((1<<i)>delta)
		continue;
		delta-=1<<i;
		uu=fa[uu][i];
	}
	return uu;
}
void lca(int uuu,int vvv)
{
	int U=uuu;
	int V=vvv;
	int lc=0;	
	if(depth[uuu]!=depth[vvv])
	{
		lc+=depth[uuu]-depth[vvv];
		U=go_up(uuu,depth[uuu]-depth[vvv]);		
	}
	if(U==V)
	{
		ans=min(ans,lc);
		return;
	}
	for(int i=17;i>=0;i--)
	{
		if(fa[U][i]!=fa[V][i])
		{
			U=fa[U][i];
			V=fa[V][i];
			lc+=(1<<i)*2;			
		}
	}
	lc+=2;
	ans=min(lc,ans);	
}
void bfs()
{
	while(!q.empty())
	q.pop();
	for(int i=1;i<=rnt;i++)
	{
		q.push(red[i]);
		dis[red[i]]=0;
	}
	while(!q.empty())
	{
		cs=q.front();
		q.pop();
		for(int i=head[cs];e[i].u==cs;i=e[i].next)
		{
			int v=e[i].v;
			if(dis[v]>dis[cs]+1)
			{
				dis[v]=dis[cs]+1;
				q.push(v);
			}
		}
	}	
}
int main()
{
	readdata();
	memset(dis,0x3f3f3f3f,sizeof(dis));
	depth[1]=1;
	red[++rnt]=1;
	dfs(1);
	int uu;int bz;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&bz,&uu);
		if(bz==1)
		{
			red[++rnt]=uu;
			if(rnt>=100)
			{
				bfs();
				rnt=0;
			}		
			continue;
		}
		ans=dis[uu];
		for(int j=1;j<=rnt;j++)
		if(depth[uu]>depth[red[j]])
		lca(uu,red[j]);
		else lca(red[j],uu);
		printf("%d\n",ans);
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值