hdu5102 The K-th Distance bfs+结构体

题意：n个节点的一棵树，任意两节点间的距离是两点间路径的长度，因此有n*(n-1)/2个长度，求前k个的和。

(2≤n≤100000,0≤K≤min(n(n−1)/2,1000000)

分析：因为k不大，直接用bfs求出前k条路径就ok了，bfs的用法很好，用结构体记录节点，父节点，长度，找到k条后就退出，只需入队k次。这样每条路径都会算重复，因此k先乘2，结果在除2。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<ostream>
#include<istream>
#include<algorithm>
#include<queue>
#include<string>
#include<cmath>
#include<set>
#include<map>
#include<stack>
#include<vector>
#define fi first
#define se second
#define pii pair<int,int>
#define inf (1<<30)
#define eps 1e-8
#define ll long long
using namespace std;
const int maxn=110005;
int n,k;
vector<int>g[maxn];
struct Node
{
    int u,fa,st;
    Node(int a,int b,int c):u(a),fa(b),st(c){}
};
queue<Node>que;
int main()
{
    int t;
    int a,b;
    scanf("%d",&t);
    while(t--) {
        while(!que.empty())
            que.pop();
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++) {
            g[i].clear();
            que.push(Node(i,-1,0));
        }
        for(int i=0;i<n-1;i++) {
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        k*=2;
        int ans=0;
        while(!que.empty() && k) {
            Node p=que.front();
            que.pop();
            int u=p.u;
            for(int i=0;i<g[u].size();i++) {
                if(g[u][i]==p.fa)
                    continue;
                if(k==0)
                    break;
                k--;
                que.push(Node(g[u][i],u,p.st+1));
                ans+=p.st+1;
            }
        }
        printf("%d\n",ans/2);
    }
    return 0;
}