[Sdoi2013] 直径

[Sdoi2013]直径

时间限制: 1 Sec  内存限制: 256 MB

题目描述

小Q最近学习了一些图论知识。根据课本，有如下定义。树：无回路且连通的无向图，每条边都有正整数的权值来表示其长度。如果一棵树有N个节点，可以证明其有且仅有N-1 条边。 路径：一棵树上，任意两个节点之间最多有一条简单路径。我们用 dis(a,b)
表示点a和点b的路径上各边长度之和。称dis(a,b)为a、b两个节点间的距离。  
 直径：一棵树上，最长的路径为树的直径。树的直径可能不是唯一的。 
现在小Q想知道，对于给定的一棵树，其直径的长度是多少，以及有多少条边满足所有的直径都经过该边。 

输入

第一行包含一个整数N，表示节点数。 
接下来N-1行，每行三个整数a, b, c ，表示点 a和点b之间有一条长度为c
的无向边。 

输出

  
共两行。第一行一个整数，表示直径的长度。第二行一个整数，表示被所有
直径经过的边的数量。 

样例输入

 
6 
3  1 1000
1  4 10
4  2 100
4  5 50
4  6 100

样例输出

1110 
2 
 
【样例说明】 
直径共有两条，3 到2的路径和3到6的路径。这两条直径都经过边(3, 1)和边(1, 4)。

提示

对于100%的测试数据：2≤N≤200000，所有点的编号都在1..N的范围内。

Solution:
　　　　自己瞎yy了一下，然后就过了，很懵逼。直径很好求，求完以后直接记录一下直径上的点然后检查一下是否还有等长的就好。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define int long long
#define N 200005
#define root 1
int read() {
    int s=0,f=1;
    char ch=getchar();
    for( ; ch<'0'||ch>'9'; f=(ch=='-')?-1:f,ch=getchar()) ;
    for( ; ch>='0'&&ch<='9'; s=s*10+(ch^48),ch=getchar()) ;
    return s*f;
}
int n,tot,r[N],Fa[N],low,high,Dis,deep[N],dis[N],Ans2;
struct oo {int fr,to,next,vv;} c[N<<1|1];
bool vis[N],Up[N],Down[N];
void add(int x,int y,int z) {
    c[++tot].fr=x;
    c[tot].to=y;
    c[tot].vv=z;
    c[tot].next=r[x];
    r[x]=tot;
}
void dfs1(int u,int f,int d) {
    if(d>Dis) Dis=d,low=u;
    for(int i=r[u]; ~i; i=c[i].next) if(c[i].to!=f) dfs1(c[i].to,u,d+c[i].vv);
}
void dfs2(int u,int f) {
    if(deep[u]>Dis) Dis=deep[u],high=u;
    for(int i=r[u]; ~i; i=c[i].next) {
        if(c[i].to!=f) {
            Fa[c[i].to]=u,dis[c[i].to]=dis[u]+1;
            deep[c[i].to]=deep[u]+c[i].vv;
            dfs2(c[i].to,u);
        }
    }
}
void dfs(int u,int f,int d,int &mx) {
    mx=max(mx,d);
    for(int i=r[u]; ~i; i=c[i].next) if(c[i].to!=f&&!vis[c[i].to]) dfs(c[i].to,u,d+c[i].vv,mx);
}
signed main() {
    n=read();
    memset(r,0xff,sizeof(r));
    for(int x,y,z,i=1; i<n; ++i) {
        x=read(),y=read(),z=read();
        add(x,y,z),add(y,x,z);
    } dfs1(root,root,0); Dis=0; dfs2(low,low);
    for(int i=high; i!=low; i=Fa[i]) vis[i]=true;
    vis[low]=true;
    for(int i=high,mx=0; ; i=Fa[i],mx=0) {
        dfs(i,i,0,mx);
        if(mx==deep[i]) Up[i]=true;
        if(mx==Dis-deep[i]) Down[i]=true;
        if(i==low) break;
    } int pos1=0,pos2=0;
    for(int i=high; ; i=Fa[i]) {
        if(Up[i]&&!pos1) pos1=i;
        if(Down[i]) pos2=i;
        if(i==low) break;
    } Ans2=dis[pos2]-dis[pos1];
    if(Ans2<0) Ans2=0;
    printf("%lld\n%lld",Dis,Ans2);
    return 0;
}