根节点唯一

Problem - 1843D - Codeforces

#include "bits/stdc++.h"
#define int long long
using namespace std;
const int N=2e6+10;
int e[N],ne[N],h[N],de[N];
int idx,val[N];
void add(int x,int y)
{
	e[idx]=y;
	ne[idx]=h[x];
	h[x]=idx++;
	de[x]++;
}
void dfs(int u,int f)
{
	int v;
	if(de[u]==1&&f!=0)
	{
		val[u]=1;
		return;
	}
	for(int i=h[u];i!=-1;i=ne[i])
	{
		v=e[i];
		if(v==f)
		continue;
		dfs(v,u);
		val[u]+=val[v];
	}
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	int t;
	cin>>t;
	while(t--)
	{

	int n;int u,v;
	cin>>n;
for(int i=1;i<=n*2;i++)
{
	e[i]=ne[i]=de[i]=val[i]=0;
	h[i]=-1;
}

	for(int i=1;i<n;i++)
	{
		
		cin>>u>>v;
		add(u,v);
		add(v,u);
	}
	dfs(1,0);
	int q;
	cin>>q;
	for(int i=1;i<=q;i++)
	{
		cin>>u>>v;
		int sum=val[u]*val[v];
		cout<<sum<<'\n';
	}
	}
}