HDU - 3848 CC On The Tree [多端BFS]_hdu 3848 cc on the tree-优快云博客

本文介绍了一种算法，用于解决在一个有N个节点的树形结构中，找到从任意节点出发，以每秒一米的速度，最快摘到两个苹果所需的时间。通过优先队列记录所有叶子节点并进行广度优先搜索，直到找到两个苹果为止。

CC lives on the tree which has N nodes.On every leaf of the tree there is an apple(leaf means there is only one branch connect this node ) .Now CC wants to get two apple ,CC can choose any node to start and the speed of CC is one meter per second.
now she wants to know the shortest time to get two apples;
Input
Thers are many cases;
The first line of every case there is a number N(2<=N<=10000)
if n is 0 means the end of input.
Next there is n-1 lines,on the i+1 line there is three number ai,bi,ci
which means there is a branch connect node ai and node bi.
(1<=ai, bi<=N , 1<=ci<=2000)
ci means the len of the branch is ci meters ;
Output
print the mininum time to get only two apple;
Sample Input
7
1 2 1
2 3 2
3 4 1
4 5 1
3 6 3
6 7 4
0
Sample Output
5

题解

把所有叶子节点加入优先队列(优先距离叶子近的)
然后bfs
直到相遇
然后取最小答案

复杂度 $o(n\cdot \log n)$

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

const int MAXS=10*1024*1024;
char buf[MAXS],*ch;

void read(int &x){
    while(*(++ch)<32);
    for(x=0;*ch>='0';++ch) x=x*10+*ch-'0';
}

/*-------------------------------------------*/
const int INF=0x3f3f3f3f;
const int MAXN=1e4+5;
struct node{int to,cost,next;};
node dat[MAXN<<1]; int u;
int cnt[MAXN],pit[MAXN],dp[MAXN],flag[MAXN];

struct ND{int val,from,v;};
bool operator < (const ND &x,const ND &y){return x.val>y.val;}

int main()
{
    fread(buf,1,MAXS,stdin);
    ch=buf-1;
    int n;
    do{
        read(n);if(!n) break;
        memset(cnt ,0 ,sizeof(int)*(n+1));
        memset(pit ,0 ,sizeof(int)*(n+1));
        memset(dp ,-1 ,sizeof(int)*(n+1));
        u=1;
        for(int i=1;i<n;i++){
            int x,y,c;
            read(x);read(y);read(c);
            dat[u]=(node){y,c,pit[x]};
            pit[x]=u++;
            dat[u]=(node){x,c,pit[y]};
            pit[y]=u++;

            cnt[x]++;
            cnt[y]++;
        }
        priority_queue<ND> que;
        for(int i=1;i<=n;i++){
            if(cnt[i]==1){
                que.push((ND){0,0,i});
            }
        }

        int ans=INF;
        while(!que.empty()){
            ND t=que.top();que.pop();
            int c=t.val;
            int v=t.v;
            int from=t.from;
            if(dp[v]==-1){
                dp[v]=c;
                flag[v]=(from==0?v:flag[from]);
            }else {
                if(flag[from]!=flag[v])
                    ans=min(ans,c+dp[v]);
                continue;
            }
            for(int i=pit[v];i;i=dat[i].next){
                node &e=dat[i];
                if(flag[v]!=flag[e.to]){
                    que.push((ND){dp[v]+e.cost,v,e.to});
                }
            }
        }
        printf("%d\n",ans);
    }while(true);
    return 0;
}