倍增LCA+树上差分,只AC了一个点,其他全是TLE,大佬求助
查看原帖
倍增LCA+树上差分,只AC了一个点,其他全是TLE,大佬求助
509923
SongShouqian楼主2022/5/28 14:55

求助!

代码

#include<bits/stdc++.h>
using namespace std;
const int N=50005,LOG=20;
struct edge{
	int to,last;
}e[N<<2];
int head[N<<2],cnt=0,depth[N],hop[N][LOG],sum[N],ans=0;
int n,k;
void add(int u,int v)
{
	e[cnt].to=v;
	e[cnt].last=head[u];
	head[u]=cnt;
	cnt++;
}
void dfs(int x,int f,int d)
{
	depth[x]=d;
	hop[x][0]=f;
	int v;
	for(int i=head[x];i;i=e[i].last)
	{
		v=e[i].to;
		if(v==f)
		{
			continue;
		}
		dfs(v,x,d+1);
	}
}
void init()
{
	dfs(1,0,1);
	for(int i=1;i<LOG;i++)
	{
		for(int j=1;j<=n;j++)
		{
			hop[j][i]=hop[hop[j][i-1]][i-1];
		}
	}
}
void du(int &u,int h)
{
	for(int i=0;h;i++)
	{
		if(h&1)
		{
			u=hop[u][i];
		}
		h>>=1;
	}
}
int lca(int u,int v)
{
	if(depth[u]!=depth[v])
	{
		if(depth[u]<depth[v])
		{
			swap(u,v);
		}
		du(u,depth[u]-depth[v]);
	}
	if(u==v)
	{
		return u;
	}
	for(int i=LOG-1;i>=0;i++)
	{
		if(hop[u][LOG]==hop[v][LOG])
		{
			continue;
		}
		u=hop[u][i];
		v=hop[v][i];
	}
	return hop[u][0];
}
void query(int x,int f)
{
	int v;
	for(int i=head[x];i;i=e[i].last)
	{
		v=e[i].to;
		if(v==f)
		{
			continue;
		}
		query(v,x);
		sum[x]+=sum[v];
	}
	ans=max(ans,sum[x]);
}
int main()
{
	scanf("%d%d",&n,&k);
	int u,v;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	init();
	int l;
	for(int i=1;i<=k;i++)
	{
		scanf("%d%d",&u,&v);
		l=lca(u,v);
		sum[u]++;
		sum[v]++;
		sum[l]--;
		if(l>1)
		{
			sum[hop[l][0]]--;
		}
	}
	query(1,0);
	printf("%d",ans);
	return 0;
}

测试点信息

测试点编号评测结果
#1AC
#2TLE
#3TLE
#4TLE
#5TLE
#6TLE
#7TLE
#8TLE
#9TLE
#10TLE
#11TLE
#12TLE
#13TLE
#14TLE
#15TLE
2022/5/28 14:55
加载中...