当前位置: 代码迷 >> 综合 >> CF14D Two Paths (树的直径)
  详细解决方案

CF14D Two Paths (树的直径)

热度:22   发布时间:2023-12-05 21:39:06.0

Problem - 14D - Codeforces

题意:

求一棵树上两条不相交直径的最大乘积。

题解:

由于n<200,考虑枚举每个点断开的情况,从断开的两端求直径。

代码:

/*keep on going and never give up*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define ll long long
#define fast std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
inline int read()
{int x=0,k=1; char c=getchar();while(c<'0'||c>'9'){if(c=='-')k=-1;c=getchar();}while(c>='0'&&c<='9')x=(x<<3)+(x<<1)+(c^48),c=getchar();return x*k;
}
const double E = exp(1);
const double PI = acos(-1.0);
const int maxn=4e5+10;
const int Mod=1e9+7;
int n,tot,d,head[maxn],rt,d1[maxn],d2[maxn];
struct QwQ{int nxt,v,id;
}e[maxn];
void add(int x,int y,int id){e[++tot].v=y;e[tot].id=id;e[tot].nxt=head[x];head[x]=tot;
}
void dfs(int x,int fa,int id){for(int i=head[x];i;i=e[i].nxt){int v=e[i].v;if(v==fa) continue;if(e[i].id==id) {rt=e[i].v;continue;}dfs(v,x,id);if(d1[x]<d1[v]+1) {d2[x]=d1[x];d1[x]=d1[v]+1;}else if(d1[v]+1>d2[x]) d2[x]=d1[v]+1;d=max(d,d1[x]+d2[x]);}
}
string s; 
signed main(){n=read();for(int i=1;i<n;i++){int u=read(),v=read();add(u,v,i);add(v,u,i);	}int ans=-1;for(int i=1;i<n;i++){d=0;rt=0;dfs(1,0,i);memset(d1,0,sizeof(d1));memset(d2,0,sizeof(d2));int l1=d;d=0;dfs(rt,0,i);memset(d1,0,sizeof(d1));memset(d2,0,sizeof(d2));ans=max(ans,l1*d);//  cout<<l1<<" "<<d<<endl;}cout<<ans<<endl;
}

  相关解决方案