BZOJ 3124 [Sdoi2013]直径 DFS

题意:

给定一棵树，求直径长度以及所有直径经过的边的个数。

---

解析:

第一问SB题，上树形DP或者任意挑选一个点，搜离他最远的点，再从该点搜离该点最远的点都可以。
我选择了后者。
因为第二问可以用到。
首先被所有直径都经过的边一定在一条直径上，且一定是连续的。
所以我们拿第一问随便搜出来的一条直径，扫描其上的所有点。
对于每一个点，我们搜离该点最远的点，并且不能够经过我们搜出来的这条直径。
那么如果这段距离等于该点到我们搜出来的直径的一个端点的
距离的话，那么该点到这个端点之间的所有边都不能称为答案。
因为我们无法走直径上的边，所以这个搜索的复杂度是O(n)，
在加上第一问我们深搜两边的O(n)复杂度。
整体复杂度O(n)。

---

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 200100
using namespace std;
typedef long long ll;
int n,cnt;
int head[N];
struct node
{
    int from,to,next;
    ll val;
}edge[N<<1];
void init()
{
    memset(head,-1,sizeof(head));
    cnt=1;
}
void edgeadd(int from,int to,ll val)
{
    edge[cnt].from=from,edge[cnt].to=to,edge[cnt].val=val;
    edge[cnt].next=head[from];
    head[from]=cnt++;
}
ll madis;
int no1,no2;
int f[N];
int vis[N];
void dfs1(int now,int ff,ll dis)
{
    if(madis<dis)
    {
        madis=dis;
        no1=now;
    }
    for(int i=head[now];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to==ff)continue;
        dfs1(to,now,dis+edge[i].val);
    }
}
void dfs2(int now,int ff,ll dis)
{
    if(madis<dis)
    {
        madis=dis;
        no2=now;
    }
    for(int i=head[now];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to==ff)continue;
        f[to]=i;
        dfs2(to,now,dis+edge[i].val);
    }
}
ll newdis;
void dfs3(int now,int ff,ll dis)
{
    if(newdis<dis)
    {
        newdis=dis;
    }
    for(int i=head[now];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        if(to==ff||vis[to])continue;
        dfs3(to,now,dis+edge[i].val);
    }
}
int main()
{
    init();
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int u,v;
        ll val;
        scanf("%d%d%lld",&u,&v,&val);
        edgeadd(u,v,val),edgeadd(v,u,val);
    }
    madis=-1;dfs1(1,0,0);
    madis=-1;dfs2(no1,0,0);
    printf("%lld\n",madis);
    f[no1]=-1;
    int cntgeshu=0;
    for(int i=f[no2];i!=-1;i=f[edge[i].from])
    {cntgeshu++;vis[edge[i].to]=1;}
    vis[no1]=1;
    int newcnt=0;
    ll nowdis=0;
    int l,r;
    for(int i=f[no2];i!=-1;i=f[edge[i].from])
    {
        newcnt++;
        nowdis+=edge[i].val;
        newdis=0;
        dfs3(edge[i].from,0,0);
        if(newdis==nowdis)l=newcnt;
        if(newdis==madis-nowdis)break;
    }
    printf("%d\n",newcnt-l);
}