蓝书（算法竞赛进阶指南）刷题记录——【SDOI2013】BZOJ3124 直径题解（树的直径）

题目：BZOJ3124.
题目大意：给定一棵$n$个点的树，求这棵树的直径以及所有树直径都经过的边的数量.
$1\le n\le 2\ast 10^5$.

首先第一问很好解决.

对于第二问，一棵树上所有直径必然有至少一个交点，并且所有直径的交必然是一条路径，考虑利用这个性质求解.

首先跑出一条路径具体的路径，考虑在这条路径上跑出一个区间作为答案.

考虑一个性质，设直径的两个端点为$x,y$，那么对于一个点$k$，若$x$到$k$的距离与以$k$为一个端点不经过直径的最长链长度之和等于直径长度，那么显然$k$到$y$这一段上的所有边都不是必须边.

利用这个性质，我们就可以直接得到直径交的两个端点了.

时间复杂度$O(n)$.

代码如下：

#include<bits/stdc++.h>
using namespace std;
 
#define Abigail inline void
typedef long long LL;
 
const int N=200000;
const LL INF=(1LL<<60)-1;
 
int n;
struct side{
  int y,next,v;
}e[N*2+9];
int lin[N+9],cs;
 
void Ins(int x,int y,int v){e[++cs].y=y;e[cs].v=v;e[cs].next=lin[x];lin[x]=cs;}
void Ins2(int x,int y,int v){Ins(x,y,v);Ins(y,x,v);}
 
LL dp[N+9][2],ans0;
int pre[N+9][2];
 
void Update(int k,int id){
  LL v=dp[e[id].y][0]+e[id].v;
  if (v>dp[k][0]){
    dp[k][1]=dp[k][0];pre[k][1]=pre[k][0];
    dp[k][0]=v;pre[k][0]=id;
  }else if (v>dp[k][1]) dp[k][1]=v,pre[k][1]=id;
}
 
void Dfs_dp(int k,int fa){
  dp[k][0]=0;dp[k][1]=-INF;
  for (int i=lin[k];i;i=e[i].next)
    if (e[i].y^fa){
      Dfs_dp(e[i].y,k);
      Update(k,i);
    }
  ans0=max(ans0,dp[k][0]+dp[k][1]);
}
 
int dia[N+9],cd;
LL v[N+9];
 
void Get_dia(){
  Dfs_dp(1,0);
  int mx=1;
  for (int i=1;i<=n;++i)
    if (dp[i][0]+dp[i][1]>dp[mx][0]+dp[mx][1]) mx=i;
  int k=mx;
  for (;k;k=e[pre[k][0]].y) dia[++cd]=k,v[cd]=e[pre[k][0]].v;
  reverse(dia+1,dia+cd+1);
  reverse(v+1,v+cd+1);
  v[cd+1]=e[pre[mx][1]].v;
  for (k=e[pre[mx][1]].y;k;k=e[pre[k][0]].y) dia[++cd]=k,v[cd+1]=e[pre[k][0]].v;
}
 
LL dis[N+9];
int vis[N+9];
 
void Dfs_dis(int k,int id,LL vd){
  vis[k]=1;
  if (vd>dis[id]) dis[id]=vd;
  for (int i=lin[k];i;i=e[i].next)
    if (!vis[e[i].y]) Dfs_dis(e[i].y,id,vd+e[i].v);
}
 
void Get_dis(){
  for (int i=1;i<=cd;++i) vis[dia[i]]=1;
  for (int i=1;i<=cd;++i) Dfs_dis(dia[i],i,0);
}
 
int ans1;
 
void Get_ans1(){
  for (int i=1;i<=cd;++i) v[i]+=v[i-1];
  int l=cd,r=1;
  for (;r<=cd;++r)
    if (dis[r]+v[r]==ans0) break;
  for (;l>=1;--l)
    if (dis[l]+v[cd]-v[l]==ans0) break;
  ans1=r-l;
}
 
Abigail into(){
  scanf("%d",&n);
  for (int i=1;i<n;++i){
    int x,y,v;
    scanf("%d%d%d",&x,&y,&v);
    Ins2(x,y,v);
  }
}
 
Abigail work(){
  Get_dia();
  Get_dis();
  Get_ans1();
}
 
Abigail outo(){
  printf("%lld\n%d\n",ans0,ans1);
}

int main(){
  into();
  work();
  outo();
  return 0;
}