题意:给你n-1条边,然后没两个节点的距离按照递增的顺序,求出前k项的和。
官方题解:
把所有边(u,v) 以及(v,u)放入一个队列,队列每弹出一个元素(u,v),对于所有与u相邻的点w,如果w!=v,就把(w,u)入队。这样就能一个一个生成前K小的距离。 注意到每条边实际上会入队两次,只要把K翻倍且把ans除2即可,时间复杂度为O(n+K);
这里只是实现一下而已。
代码:
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#pragma comment(linker, "/STACK:1024000000,1024000000") #include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <cmath> #include <algorithm> #include <queue> using namespace std; #define N 100007 struct node { int u,v,d; node(int _u,int _v,int _d):u(_u),v(_v),d(_d){} node(){} }; struct Edge{ int v,next; }G[2*N]; int ans,k,tot,head[N]; queue<node> q; void addedge(int u,int v) { G[tot].v = v; G[tot].next = head[u]; head[u] = tot++; } void bfs() { int cnt = 0; while(!q.empty()) { node tmp = q.front(); q.pop(); int u = tmp.u, v = tmp.v, d = tmp.d; if(cnt >= k) break; for(int i=head[u];i!=-1;i=G[i].next) { int vv = G[i].v; if(vv != v) { ans += d+1; cnt++; q.push(node(vv,u,d+1)); } if(cnt >= k) break; } if(cnt >= k) break; } } int main() { int t,n,u,v,i; scanf("%d",&t); while(t--) { while(!q.empty()) q.pop(); scanf("%d%d",&n,&k); tot = 0; memset(head,-1,sizeof(head)); for(i=1;i<=n;i++) q.push(node(i,i,0)); for(i=1;i<n;i++) { scanf("%d%d",&u,&v); addedge(u,v); addedge(v,u); } k *= 2; ans = 0; bfs(); cout<<ans/2<<endl; } return 0; }