bzoj1509[NOI2003] 逃学的小孩

题目链接：bzoj1509
题目大意：
给你一棵树。让你找三个点x,y,z。求从一个点x出发，先到另外两个点中距离x较近的点(设为)y，再到剩下的那个点z，的时间。即max(dis(x,y),dis(y,z))。

题解：
树的直径。理应是树形dp，dfs也可以。
最坏情况下就是花得时间最长。那么直径(u,v)一定含在路径中。然后再选一个min(dis(u,x),dis(v,u))最大的点加上直径就是答案。
虽说是treedp但是我直接打了n个dfs了。感觉方便点啊。

怎么求直径：
随便从一个点出发到离他最远的点u,该点一定是直径的一端。再从u出发到离他最远的点v。u->v就是该树的直径。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define maxn 200010

struct node
{
    int x,y,next;LL c;
}a[maxn*2];int len,first[maxn];
LL ans,d[maxn];int u,v,fa[maxn];bool mk[maxn];
LL mymax(LL x,LL y){return (x>y)?x:y;}
LL mymin(LL x,LL y){return (x<y)?x:y;}
void ins(int x,int y,LL c)
{
    len++;a[len].x=x;a[len].y=y;a[len].c=c;
    a[len].next=first[x];first[x]=len;
}
int dfs(int x)
{
    int as=x;
    for (int k=first[x];k!=-1;k=a[k].next)
    {
        int y=a[k].y;
        if (y==fa[x]) continue;
        d[y]=d[x]+a[k].c;fa[y]=x;
        int p=dfs(y);
        if (d[p]>d[as]) as=p;
    }
    return as;
}
void mark(int x)
{
    while (x!=u)
    {
        mk[x]=true;
        x=fa[x];
    }mk[x]=true;
}
LL godown(int x)
{
    LL ret=0;
    for (int k=first[x];k!=-1;k=a[k].next)
    {
        int y=a[k].y;
        if (y==fa[x]) continue;
        ret=mymax(ret,godown(y)+a[k].c);
    }return ret;
}
void fan(int x)
{
    if (x!=u && x!=v) ans=mymax(ans,mymin(d[x],d[v]-d[x]));
    for (int k=first[x];k!=-1;k=a[k].next)
    {
        int y=a[k].y;
        if (y==fa[x]) continue;
        if (!mk[y]) ans=mymax(ans,a[k].c+godown(y)+mymin(d[x],d[v]-d[x]));
        else fan(y);
    }
}
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int n,i,m,x,y;LL c;
    scanf("%d%d",&n,&m);
    len=0;memset(first,-1,sizeof(first));
    for (i=1;i<=m;i++)
    {
        scanf("%d%d%lld",&x,&y,&c);
        ins(x,y,c);ins(y,x,c);
        mk[i]=false;
    }mk[n]=false;
    fa[1]=0;d[1]=0;u=dfs(1);
    fa[u]=0;d[u]=0;v=dfs(u);
    mark(v);ans=0;fan(u);
    ans+=d[v];printf("%lld\n",ans);
    return 0;
}