[SDOI2013] 直径

SDOI2013 直径

题目描述

小Q最近学习了一些图论知识。根据课本，有如下定义。树：无回路且连通的无向图，每条边都有正整数的权值来表示其长度。如果一棵树有N个节点，可以证明其有且仅有N-1 条边。

路径：一棵树上，任意两个节点之间最多有一条简单路径。我们用 dis(a,b)表示点a和点b的路径上各边长度之和。称dis(a,b)为a、b两个节点间的距离。

直径：一棵树上，最长的路径为树的直径。树的直径可能不是唯一的。

现在小Q想知道，对于给定的一棵树，其直径的长度是多少，以及有多少条边满足所有的直径都经过该边。

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：

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

输出样例#1：

1110
2

说明

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

对于100%的测试数据：2<=N<=200000，所有点的编号都在1..N的范围内，边的权值<=10^9。

题解

题目传送门:SDOI2013
这道题思路不难,代码也简单(真的不是一般的难调)
讲一下思路吧,

对于第一问,两遍dfs/bfs/dp就出来了,但是由于我们后面还需要用到直径这条路径,所以用dfs/bfs比较好

第二问
如何判断一条边是否是直径上的必须边
我们知道,树的直径长度只有一个,但不只一条
设一条直径上一个点u,我们发现若以u为根的子树,不包括直径上的节点,离它最远的点距离与直径上的端点到它的距离一样,这说明有另一条直径
举个例子,我们看下面一幅图

可以看到2-9-8-5-6-1是它的一条直径
但是仔细观察就可以发现,它不只这一条直径

我们看到上面这幅图,3-5-6-1或者是3-5-6-4-10也是一条,还有其他的

所以我们可以保存直径的两个端点,然后左端点向右跳,右端点向左跳,看着条路径上有没有满足上述条件的点,如果有,就把左短点/右端点移到这个点

Code

#include<bits/stdc++.h>
#define rg register
#define Min(a,b) (a)<(b)?(a):(b)
#define Max(a,b) (a)>(b)?(a):(b)

using namespace std;
typedef long long lol;

const int N=200010;

void in(int &ans)
{
    ans=0;int f=1;char i=getchar();
    while(i<'0' || i>'9') {if(i=='-') f=-1;i=getchar();}
    while(i>='0' && i<='9') ans=(ans<<1)+(ans<<3)+(i^48),i=getchar();
    ans*=f;
}

int n,m,cnt,u,v,ans,B;
int nex[N<<1],to[N<<1],head[N],w[N<<1];
int son[N],vis[N],fa[N],dep[N];
lol dis[N],f[N];

inline void add(int a,int b,int c)
{
    to[++cnt]=b,nex[cnt]=head[a];
    w[cnt]=c,head[a]=cnt;
}

void dfs(int u)
{
    for(int i=head[u];i;i=nex[i])
        if(to[i]!=fa[u])
            fa[to[i]]=u, dep[to[i]]=dep[u]+1,dis[to[i]]=dis[u]+w[i], dfs(to[i]);
}

void find(int u)
{
    f[u]=dis[u];
    for(int i=head[u];i;i=nex[i])
        if(!vis[to[i]] && to[i]!=fa[u])
            find(to[i]),f[u]=Max(f[u],f[to[i]]);
}

void Pre()
{
    int x=fa[v];
    while(x) {
        find(x);//以x为根节点的子树的最远距离=f[x]-dis[x]
        if(f[x]==dis[B]) v=x;//根据直径的最长性,从后面往前面枚举,只可能是短的那一节
        x=fa[x];
    }
}

void Next()
{
    int x=son[u];
    while(x) {
        find(x);
        if(f[x]-dis[x]==dis[x]) u=x;//同上
        x=son[x];
    }
}

void init()
{
    dfs(1);
    for(int i=1;i<=n;i++) if(dis[i]>dis[u]) u=i;
    memset(dep,0,sizeof(dep));
    memset(fa,0,sizeof(fa));
    memset(dis,0,sizeof(dis));
    dfs(u);
    for(int i=1;i<=n;i++) if(dis[i]>dis[v]) B=v=i;
    printf("%lld\n",dis[v]);
    int x=v; while(x) vis[x]=1,son[fa[x]]=x,x=fa[x];
}

int main()
{
    int a,b,c;in(n);
    for(int i=1;i<n;i++) {
        in(a),in(b),in(c);
        add(a,b,c),add(b,a,c);
    }
    init(); Next(); Pre();
    printf("%d\n",dep[v]-dep[u]);
}

博主蒟蒻,随意转载.但必须附上原文链接

http://www.cnblogs.com/real-l/