hdu 5102 The K-th Distance

http://acm.hdu.edu.cn/showproblem.php?pid=5102

补题.

题意：给你n-1条边，然后没两个节点的距离按照递增的顺序，求出前k项的和。

思路：把所有边(u,v) 以及(v,u)放入一个队列，队列每弹出一个元素(u,v)，对于所有与u相邻的点w，如果w!=v，就把(w,u)入队。这样就能一个一个生成前K小的距离。 注意到每条边实际上会入队两次，只要把K翻倍且把ans除2即可，时间复杂度为O(n+K)；

#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

int t,n,k;
struct node
{
    int u,v,w;
    node(int u,int v,int w):u(u),v(v),w(w){}
};

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        queue<node>q;
        vector<int>g[100001];
        scanf("%d%d",&n,&k);
        k*=2;
        for(int i=1; i<=n-1; i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        for(int i=1; i<=n; i++)
        {
            q.push(node(i,i,0));
        }
        int cnt=0; long long ans=0;
        while(cnt<k)
        {
            node e=q.front(); q.pop();
            int u=e.u;
            for(int j=0; j<(int)g[u].size(); j++)
            {
                int v=g[u][j];
                if(v!=e.v&&cnt<k)
                {
                    q.push(node(v,u,e.w+1));
                    cnt++;
                    ans+=e.w+1;
                }
            }
        }
        printf("%lld\n",ans/2);
    }
    return 0;
}