[bzoj3124][乱搞]直径

Description

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

Input

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

Output

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

Sample Input

6

3 1 1000

1 4 10

4 2 100

4 5 50

4 6 100

Sample Output

1110

2

HINT

【样例说明】

直径共有两条，3 到2的路径和3到6的路径。这两条直径都经过边(3, 1)和边(1, 4)。 HINT

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

边的权值≤10^9。

题解

第一问不用说了吧。。
关于第二问，有两个性质
1：满足答案的边都在一条直径上（因为每条直径上面都有嘛）
2：这些边一定是连续的一条链，这个yy一下就知道了的。。
如果直径上有一个点，这个点能引出两条或以上的相同长度且最长的链，那么这个点之后的所有点都不可能成为答案
于是我们可以跑两遍dfs，同时记录答案
然后就没了。。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
struct node
{
    int x,y,next,other;LL c;
}a[410000];int len,last[210000];
void ins(int x,int y,LL c)
{
    int k1,k2;
    k1=++len;
    a[len].x=x;a[len].y=y;a[len].c=c;
    a[len].next=last[x];last[x]=len;
    k2=++len;
    a[len].x=y;a[len].y=x;a[len].c=c;
    a[len].next=last[y];last[y]=len;
    a[k1].other=k2;a[k2].other=k1;
}
int G,lx,ly;
LL d[210000];
int n;
void getmx(int x,int fa)
{
    if(d[x]>d[G])G=x;
    for(int k=last[x];k;k=a[k].next)if(a[k].y!=fa)d[a[k].y]=d[x]+a[k].c,getmx(a[k].y,x);
}
int pre[210000],nxt[210000],tmp[210000];
bool bk;
void getpath(int x,int ed,int fa)
{
    if(x==ed){bk=true;return ;}
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa)
        {
            pre[y]=x;tmp[y]=k;
            getpath(y,ed,x);
        }
        if(bk)return;
    }
}
bool vis[210000];
LL f[210000];
void dfs(int x,int fa)
{
    f[x]=0;bool bk=false;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa)
        {
            dfs(y,x);LL sum=f[y]+a[k].c;
            if(f[x]>sum)continue;
            if(f[x]<sum){bk=false;f[x]=sum;}
            else bk=true;
        }
    }
    if(bk==true)vis[x]=true;//这里开始有两个相等的最长 
}
bool v[210000];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y;LL c;scanf("%d%d%lld",&x,&y,&c);
        ins(x,y,c);
    }
    G=0;getmx(1,0);
    lx=G;d[lx]=0;G=0;
    getmx(lx,0);
    ly=G;
    printf("%lld\n",d[ly]);
    bk=false;getpath(lx,ly,0);
    memset(v,false,sizeof(v));
    int u=ly;
    while(u!=lx)nxt[pre[u]]=u,u=pre[u];
    memset(vis,false,sizeof(vis));
    dfs(lx,0);
    u=lx;
    while(u!=ly)
    {
        if(!vis[u])v[u]=true;
        else {v[u]=true;break;}
        u=nxt[u];
    }
    if(u==ly && !vis[u])v[u]=true;
    memset(vis,false,sizeof(vis));
    dfs(ly,0);
    u=ly;
    int ans=0;
    while(u!=lx)
    {
        if(!vis[u])
        {
            if(v[u])ans++;
        }
        else 
        {
            if(v[u])ans++;
            break;
        }
        u=pre[u];
    }
    if(u==lx && !vis[u] && v[u])ans++;
    printf("%d\n",ans-1);
    return 0;
}