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 200010struct 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;}