10‘,需要求助,调者关
查看原帖
10‘,需要求助,调者关
1101266
Fe_CuSO4__FeSO4_Cu楼主2025/1/17 13:49
#include <bits/stdc++.h>
using namespace std;
#define N 1000020
#define ll long long
#define bol bitset
#define imax INT_MAX
#define imin INT_MIN
#define inf 0x3f3f3f3f
#define B 100000010
#define min(a,b) (a<b ? a:b)
#define max(a,b) (a>b ? a:b)
int fa[N][20],FA[N],dep[N],n,m;
ll sum[N];
vector<int>e[N];
void dfs(int u)
{
	for(int i=0;i<int(e[u].size());i++)
	{
		int k=e[u][i];
		if(k==FA[u]) continue;
		sum[k]=int(e[k].size())+sum[u];
		FA[k]=u;
		dep[k]=dep[u]+1;
		dfs(k);
	}
}
void init()
{
	for(int i=1;i<=n;i++) fa[i][0]=FA[i];
	for(int i=1;i<=18;i++)
	{
		for(int j=1;j<=n;j++)
		{
			fa[j][i]=fa[fa[j][i-1]][i-1];
		}
	}
	
}

int LCA(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=18;i>=0;i--)
	{
		if(dep[y]<=dep[fa[x][i]]) x=fa[x][i];
	}
	if(x==y) return x; 
	for(int i=18;i>=0;i--)
	{
		if(fa[x][i]!=fa[y][i])
		{
			x=fa[x][i],y=fa[y][i];
		}
	}
	return (fa[x][0]=0 ? 1 : fa[x][0]);
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		e[u].push_back(v);
		e[v].push_back(u);
	}
	sum[1]=int(e[1].size());
	dfs(1);
	init();
	for(int i=1;i<=m;i++)
	{
		int x,y;
		cin>>x>>y;
		int f=LCA(x,y);
		cout<<sum[x]+sum[y]-2*sum[f]+int(e[f].size())<<"\n";
	}
	return 0;
}

如标题,只有10pts,其余全WA。

2025/1/17 13:49
加载中...