hihocoder1050 : 树中的最长路

时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述

上回说到,小Ho得到了一棵二叉树玩具,这个玩具是由小球和木棍连接起来的,而在拆拼它的过程中,小Ho发现他不仅仅可以拼凑成一棵二叉树!还可以拼凑成一棵多叉树——好吧,其实就是更为平常的树而已。

但是不管怎么说,小Ho喜爱的玩具又升级换代了,于是他更加爱不释手(其实说起来小球和木棍有什么好玩的是吧= =)。小Ho手中的这棵玩具树现在由N个小球和N-1根木棍拼凑而成,这N个小球都被小Ho标上了不同的数字,并且这些数字都是出于1..N的范围之内,每根木棍都连接着两个不同的小球,并且保证任意两个小球间都不存在两条不同的路径可以互相到达。总而言之,是一个相当好玩的玩具啦!

但是小Hi瞧见小Ho这个样子,觉得他这样沉迷其中并不是一件好事,于是寻思着再找点问题让他来思考思考——不过以小Hi的水准,自然是手到擒来啦!

于是这天食过早饭后,小Hi便对着又拿着树玩具玩的不亦乐乎的小Ho道:“你说你天天玩这个东西,我就问你一个问题,看看你可否知道?”

“不好!”小Ho想都不想的拒绝了。

“那你就继续玩吧,一会回国的时候我不叫上你了~”小Hi严肃道。

“诶!别别别,你说你说,我听着呢。”一向习惯于开启跟随模式的小Ho忍不住了,马上喊道。

小Hi满意的点了点头,随即说道:“这才对嘛,我的问题很简单,就是——你这棵树中哪两个结点之间的距离最长?当然,这里的距离是指从一个结点走到另一个结点经过的木棍数。”。

“啊?”小Ho低头看了看手里的玩具树,困惑了。

提示一:路总有折点,路径也不例外!
输入

每个测试点(输入文件)有且仅有一组测试数据。

每组测试数据的第一行为一个整数N,意义如前文所述。

每组测试数据的第2~N行,每行分别描述一根木棍,其中第i+1行为两个整数Ai,Bi,表示第i根木棍连接的两个小球的编号。

对于20%的数据,满足N<=10。

对于50%的数据,满足N<=10^3。

对于100%的数据,满足N<=10^5,1<=Ai<=N, 1<=Bi<=N

小Hi的Tip:那些用数组存储树边的记得要开两倍大小哦!

输出

对于每组测试数据,输出一个整数Ans,表示给出的这棵树中距离最远的两个结点之间相隔的距离。

样例输入
8
1 2
1 3
1 4
4 5
3 6
6 7
7 8
样例输出
6

import java.io.BufferedInputStream;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {

        new Task().solve();
    }

}

class Task {

    void solve() {
        Scanner cin = new Scanner(new BufferedInputStream(System.in));
        PrintWriter cout = new PrintWriter(System.out);

        while (cin.hasNext()) {
            n = cin.nextInt();
            graph = new Graph(n);
            for (int i = 1; i < n; i++) {
                int u = cin.nextInt() - 1;
                int v = cin.nextInt() - 1;
                graph.add(u, v);
                graph.add(v, u);
            }
            int[] p = bfs(0);
            p = bfs(p[0]);
            cout.println(p[1]);
        }

        cout.flush();
    }

    int n;
    Graph graph;

    int[] bfs(int start) {
        int[] dist = new int[n];
        boolean[] inq = new boolean[n];
        Arrays.fill(dist, Integer.MAX_VALUE);
        Arrays.fill(inq, false);
        dist[start] = 0;
        inq[start] = true;
        Queue<Integer> q = new LinkedList<Integer>();
        q.add(start);
        while (!q.isEmpty()) {
            int u = q.poll();
            inq[u] = false;
            for (int i = graph.head[u]; i != -1; i = graph.e[i].next) {
                int v = graph.e[i].v;
                if (dist[v] > dist[u] + 1) {
                    dist[v] = dist[u] + 1;
                    q.add(v);
                    inq[v] = true;
                }
            }
        }

        int p = 0;
        int w = dist[0];
        for (int i = 1; i < n; i++) {
            if (dist[i] > w) {
                w = dist[i];
                p = i;
            }
        }

        return new int[] { p, w };
    }

}

class Graph {

    Graph(int n) {
        head = new int[n];
        e = new E[n * 2 + 10];
        idx = 0;
        Arrays.fill(head, -1);
    }

    class E {
        int v;
        int next;
    }

    void add(int u, int v) {
        e[idx] = new E();
        e[idx].v = v;
        e[idx].next = head[u];
        head[u] = idx++;
    }

    int idx;
    E[] e;
    int[] head;

}

你可能感兴趣的:(hihocoder1050 : 树中的最长路)