标签：dep return fa -- 题解 int zhengjun find
题目大意
给定一个 \(n\) 个点的树，每个点有权值 \(v_i\)，每条边也有权值 \(w_j\)，对于树上一条简单路径，它的权值就是路径上（包括两端点）\(\min{\{v_i\}}\times\sum w_j\)，求最大的路径权值。
思路
显然可以发现可以将每个点按照 \(a_i\) 从大到小排序，枚举当前的 \(\min\{v_i\}\)，然后发现原来的树会分成若干个连通块，只需要维护出每个连通块的直径即可。
合并连通块的时候，用左边直径的两个端点，和右边直径的两个端点，一共四个点，那么合并完的直径端点必然在这些点当中，所以枚举 \(6\) 种情况即可。
代码
#include<bits/stdc++.h>
using namespace std;typedef long long ll;const int N=2e5+10,K=log2(N)+2;vector<pair<int,int> >e[N];
int n,k,a[N],head[N],kk,f[K][N],dep[N],p[N][2],t[4],fa[N],cur[N];ll d[N],mx,ans;
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
void dfs(int u){for(auto [v,w]:e[u])if(v^f[0][u])f[0][v]=u,dep[v]=dep[u]+1,d[v]=d[u]+w,dfs(v);}
void init(){for(int i=1;(1<<i)<=n;i++)for(int j=1;j<=n;j++)f[i][j]=f[i-1][f[i-1][j]];}
int LCA(int u,int v){
if(dep[u]<dep[v])swap(u,v);for(int i=k;i>=0;i--)if(dep[f[i][u]]>=dep[v])u=f[i][u];
if(u==v)return u;for(int i=k;i>=0;i--)if(f[i][u]!=f[i][v])u=f[i][u],v=f[i][v];return f[0][u];
}
ll dis(int u,int v){int fa=LCA(u,v);return d[u]+d[v]-(d[fa]<<1);}
void merge(int u,int v){
u=find(u);v=find(v);tuple<ll,int,int>h[6];t[0]=p[u][0];t[1]=p[u][1];t[2]=p[v][0];t[3]=p[v][1];
int now=0;for(int i=0;i<4;i++)for(int j=i+1;j<4;j++)h[++now]={dis(t[i],t[j]),t[i],t[j]};
h[0]=max(h[0],max(h[1],max(h[2],max(h[3],max(h[4],h[5])))));
mx=max(mx,get<0>(h[0]));p[u][0]=get<1>(h[0]);p[u][1]=get<2>(h[0]);fa[v]=u;
}
int main(){
scanf("%d",&n);k=log2(n);for(int i=1;i<=n;i++)scanf("%d",&a[fa[i]=cur[i]=p[i][0]=p[i][1]=i]);
for(int i=1,u,v,w;i<n;i++)scanf("%d%d%d",&u,&v,&w),e[u].push_back({v,w}),e[v].push_back({u,w});
dfs(1);init();sort(cur+1,cur+1+n,[](int x,int y){return a[x]>a[y];});for(int i=1,u;u=cur[i],i<=n;i++){
for(auto [v,w]:e[u])if(a[v]>=a[u]&&find(u)^find(v))merge(u,v);ans=max(ans,a[u]*mx);}cout<<ans;return 0;
}
如有问题请指出，谢谢--zhengjun
标签：dep,return,fa,--,题解,int,zhengjun,find 来源： https://www.cnblogs.com/A-zjzj/p/16366099.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。