BZOJ 3124 [Sdoi2013]直径 DFS

题意:

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

---

解析:

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

---

代码:

#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>#define N 200100using 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);}