HDU 5102 The K-th Distance 队列(树的前k路径)

HDU 5102
题意:n个结点的树,两点距离有n*(n-1)/2对,求出dis(u,v)前K小之和,n<=1e5,k<=1e6
把dis==0(每个点)的都压入队列,每次弹出队列中的一个点,然后往外延伸一条边,依次从队列中生成dis=1,2.3.....的路径即可

u-v v-to  树上无环,只要注意to!=u,生出的路径长度就是递增的

1-2,2-1被生成两次 k*2 ans/2即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+20;
struct node{
	int u,v,dis;
	node(int a,int b,int c)
	{
		u=a,v=b,dis=c;
	}
};
vector<int> e[N];
queue<node> q;
int main()
{
	int T,n,k,u,v;
	cin>>T;
	while(T--)
	{
	
		scanf("%d%d",&n,&k);
		while(!q.empty())
			q.pop();
		for(int i=1;i<=n;i++)	
			e[i].clear();

		for(int i=1;i<=n-1;i++)
		{
			scanf("%d%d",&u,&v);
			e[u].push_back(v);
			e[v].push_back(u);
		}
		for(int i=1;i<=n;i++)
			q.push(node(i,i,0));
		k*=2;
		int cnt=0;
		ll ans=0;
		while(!q.empty()&&cnt<k)
		{	
			node t=q.front();
			q.pop();	
			int u=t.u,v=t.v,d=t.dis;
			for(int i=0;i<e[v].size();i++)
			{
				int to=e[v][i];
				if(u==to)	continue;
				if(cnt<k)
				{
					cnt++,ans+=d+1;
					q.push(node(v,to,d+1));
				}	
			}
		}	
		cout<<ans/2<<endl;
	}
	return 0;
}