F. Three Paths on a Tree(树的直径)

https://codeforces.com/contest/1294/problem/F

---

思路：

思路还是比较好想的。

取两个点放到直径，必然是最优的。剩下一个算距离最大的就好。

所以开始我是把直径的点标记了。然后最后扫的时候再算一下到lca(i,A),lca(i,B)去更新最值。但是一直wa。

最后索性改成三个点的Σdis/2，啥也不判了。

还没找到原来的问题。感觉是getdis写错了...

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+3000;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
vector<LL>g[maxn];
LL A,B;
LL disa[maxn],disb[maxn],tdis[maxn];
LL fat[maxn][24];
LL dep[maxn];
bool used[maxn];
void dfs(LL u,LL fa,LL dis[]){
     dep[u]=dep[fa]+1;
     fat[u][0]=fa;
     for(LL i=1;(1<<i)<=dep[u];i++){
         fat[u][i]=fat[ fat[u][i-1] ][i-1];
     }
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa) continue;
         dis[v]=dis[u]+1;
         dfs(v,u,dis);
     }
}
LL LCA(LL x,LL y){
   if(dep[x]<dep[y]) swap(x,y);
   for(LL i=21;i>=0;i--){
      if(dep[x]-(1<<i)>=dep[y]){
         x=fat[x][i];
      }
   }
   if(x==y) return x;
   for(LL i=21;i>=0;i--){
      if(fat[x][i]!=fat[y][i]){
         x=fat[x][i];
         y=fat[y][i];
      }
   }
   return fat[x][0];
}
LL getdis(LL x,LL y){
   return (dep[x]+dep[y])-2*dep[LCA(x,y)];
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL n;cin>>n;
   for(LL i=1;i<n;i++){
      LL u,v;cin>>u>>v;
      g[u].push_back(v);
      g[v].push_back(u);
   }
   dfs(1,0,tdis);
   LL mx=-1;
   for(LL i=1;i<=n;i++){
       if(mx<tdis[i]){
          mx=tdis[i];
          A=i;
       }
   }
   memset(fat,0,sizeof(fat));///一定要清空
   dfs(A,0,disa);
   mx=-1;
   for(LL i=1;i<=n;i++){
       if(mx<disa[i]){
          mx=disa[i];
          B=i;
       }
   }
   for(LL i=B;i!=0;i=fat[i][0]){
       used[i]=true;
   }
   dfs(B,0,disb);
   LL ans=0;
   LL C=0;mx=-1;bool flag=1;
   ///LL root=LCA(A,B);
   for(LL i=1;i<=n;i++){
       LL res=(disa[B]+disa[i]+disb[i])/2;
       if(res>ans&&i!=A&&i!=B){
          ans=res;
          C=i;
       }
       ///LL res=disa[B];
     ///  if(i==A||i==B) continue;
       ///LL LCA1=LCA(i,A);LL LCA2=LCA(i,B);
      ///换一个判断是否在树的直径上的做法
//      if(used[i]){
//         if(flag) C=i,flag=0;
//         continue;
//      }
//       if(LCA1!=root){
//          LL add=getdis(i,LCA1);
//          res+=add;
//       }
//       if(LCA2!=root){
//          LL add=getdis(i,LCA2);
//          res+=add;
//       }
//       LL res=(getdis(i,A)+getdis(i,B)+getdis(A,B))/2;
//       cout<<"i="<<i<<" "<<"res="<<res<<"\n";
   }
   cout<<ans<<"\n";
   cout<<A<<" "<<B<<" "<<C<<"\n";
   return 0;
}

一直wa11的

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+1000;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
vector<LL>g[maxn];
LL A,B;
LL disa[maxn],disb[maxn],tdis[maxn];
LL fat[maxn][22];
LL dep[maxn];
bool used[maxn];
void dfs(LL u,LL fa,LL dis[]){
     dep[u]=dep[fa]+1;
     fat[u][0]=fa;
     for(LL i=1;(1<<i)<=dep[u];i++){
         fat[u][i]=fat[ fat[u][i-1] ][i-1];
     }
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa) continue;
         dis[v]=dis[u]+1;
         dfs(v,u,dis);
     }
}
LL LCA(LL x,LL y){
   if(dep[x]<dep[y]) swap(x,y);
   for(LL i=19;i>=0;i--){
      if(dep[x]-(1<<i)>=dep[y]){
         x=fat[x][i];
      }
   }
   if(x==y) return x;
   for(LL i=19;i>=0;i--){
      if(fat[x][i]!=fat[y][i]){
         x=fat[x][i];
         y=fat[y][i];
      }
   }
   return fat[x][0];
}
LL getdis(LL x,LL y){
   return (dep[x]+dep[y])-2*dep[LCA(x,y)];
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL n;cin>>n;
   for(LL i=1;i<n;i++){
      LL u,v;cin>>u>>v;
      g[u].push_back(v);
      g[v].push_back(u);
   }
   dfs(1,-1,tdis);
   LL mx=-1;
   for(LL i=1;i<=n;i++){
       if(mx<tdis[i]){
          mx=tdis[i];
          A=i;
       }
   }
   memset(fat,0,sizeof(fat));
   dfs(A,-1,disa);
   mx=-1;
   for(LL i=1;i<=n;i++){
       if(mx<disa[i]){
          mx=disa[i];
          B=i;
       }
   }
   for(LL i=B;i!=0;i=fat[i][0]){
       used[i]=true;
   }

   ///dfs(B,0,disb);?
   LL ans=disa[B];
   LL C=0;mx=-1;bool flag=1;
   LL root=LCA(A,B);
   for(LL i=1;i<=n;i++){
       LL res=disa[B];
       if(i==A||i==B) continue;
       LL LCA1=LCA(i,A);LL LCA2=LCA(i,B);
      ///换一个判断是否在树的直径上的做法
      if(used[i]){
         if(flag) C=i,flag=0;
         continue;
      }
       if(LCA1!=root){
          LL add=getdis(i,LCA1);
          res+=add;
       }
       if(LCA2!=root){
          LL add=getdis(i,LCA2);
          res+=add;
       }
       if(res>ans){
          ans=res;
          C=i;
       }
//       cout<<"i="<<i<<" "<<"res="<<res<<"\n";
   }
   cout<<ans<<"\n";
   cout<<A<<" "<<B<<" "<<C<<"\n";
   return 0;
}