利用平均数的一个性质:若每个元素都减去x,平均数也减小x,因此可以二分答案Ave,求所有边权值减去Ave后,有无含L~U条边,权值和非负的路径
-> 树的含L~U条边的最长路权值和是否>=0
点分治即可。优化:
1.先求root,再二分,可以避免多次求root
2.先只求经过root的路径的最优Ave,并把它作为 递归求不经过root的路径 时的二分下界
判断该Ave是否可行时,要得到若干不在同一子树上的链,就不从root,而从root的每个子树递归,求出每个子树v[]在每一深度下的最长路径
枚举最优路径在当前子树v[i]下的深度,若为j,则该路径在前1~i-1棵子树中,取边数(即深度)为[L-i,U-i]的最长路径
用数组mx[j]记录前1~i-1棵子树中深度为j的所有路径的最长长度
显然,当子树v[i]确定后,mx[j]数组关于j是单调的,可以用单调队列维护定长区间内的最值
给个图方便理解:
点分治算法中,树有logN层,每层中的各个点getroot,getd总时间复杂度都是O(N)的,因此算上二分答案,该算法复杂度为:O(NlogNloglim)
因此,不能在操作每个点时使用O(N)的算法,只能在操作每一层时使用均摊O(N)的算法
细节见代码中的注释
#include<stdio.h> #include<stdlib.h> double INF=1000000; double w[200005]={0},d[100005]={0},a[100005]={0},mx[100005]={0}; int v[200005]={0},first[100005]={0},next[200005]={0},size[100005]={0},maxsize[100005]={0},vis[100005]={0},q[100005]={0}; double lim=0.0,ans=0.0,ave; int n,e=0,L,U,root=0,sum,maxdeep; int min(int a,int b) { if(a<b) return a; return b; } void tj(int x,int y,double z) { v[++e]=y; w[e]=z; next[e]=first[x]; first[x]=e; } void getroot(int x,int fa) { int i; size[x]=1; maxsize[x]=0; for(i=first[x];i!=0;i=next[i]) if(v[i]!=fa&&vis[v[i]]==0) { getroot(v[i],x); size[x]+=size[v[i]]; if(maxsize[x]<size[v[i]]) maxsize[x]=size[v[i]]; } if(maxsize[x]<sum-size[x]) maxsize[x]=sum-size[x]; if(root==0||maxsize[root]>maxsize[x]) root=x; } void getd(int x,int fa,int deep) { int i; if(a[deep]<d[x]) a[deep]=d[x]; if(maxdeep<deep) maxdeep=deep; for(i=first[x];i!=0;i=next[i]) if(v[i]!=fa&&vis[v[i]]==0) { d[v[i]]=d[x]+w[i]-ave; getd(v[i],x,deep+1); } } int judge(int x) { int head,tail,i,j,k,endj,Mdeep=0; mx[0]=a[0]=d[x]=0; for(i=first[x];i!=0;i=next[i]) if(vis[v[i]]==0)//不从root递归,而从root的每个子树递归,得到若干不在同一子树上的链 { d[v[i]]=d[x]+w[i]-ave; maxdeep=0;//记录子树v[i]的最大深度 getd(v[i],x,1); endj=min(U,maxdeep);//endj:路径在子树v[i]中的深度最大值 head=0; tail=-1; for(j=min(U-1,Mdeep);j>=L;j--)//枚举子树v[i]深度前的准备:若之前mx[U-1~L]均为-INF,没必要循环一遍 { q[++tail]=j; while(tail>head&&mx[q[tail]]>mx[q[tail-1]]) { q[tail-1]=q[tail]; tail--; } } for(j=1;j<=endj;j++)//j:枚举路径在子树v[i]中的深度 { if(L-j>=0) { q[++tail]=L-j; while(tail>head&&mx[q[tail]]>mx[q[tail-1]]) { q[tail-1]=q[tail]; tail--; } } while(head<=tail&&q[head]>U-j) head++; if(a[j]+mx[q[head]]>=0) { maxdeep=0; getd(x,0,0); for(k=1;k<=Mdeep;k++)//别忘了还原mx[],a[]数组 mx[k]=a[k]=-INF; for(k=Mdeep;k<=maxdeep;k++)//别忘了还原mx[],a[]数组 mx[k]=a[k]=-INF; return 1; } } for(j=1;j<=maxdeep;j++) { if(mx[j]<a[j]) mx[j]=a[j]; a[j]=-INF; } if(Mdeep<maxdeep) Mdeep=maxdeep; } for(i=1;i<=Mdeep;i++)//mx[]改变的范围是1~maxdeep,因此还原为-INF不需要从1到n循环 mx[i]=a[i]=-INF; return 0; } void getans(int x) { double left=ans,right=lim;//二分答案,左闭右开 while(right-left>0.0001)//设置精度 { ave=(left+right)/2; if(judge(x)==1) left=ave; else right=ave; } ans=left; } void work(int x) { int i; vis[x]=1; getans(x);//求解经过root的路径的最大满足要求的Ave for(i=first[x];i!=0;i=next[i]) if(vis[v[i]]==0) { sum=size[v[i]]; root=0; getroot(v[i],x); if(size[v[i]]>L) work(root);//重要优化 } } int main() { double z; int i,x,y; INF*=INF; scanf("%d%d%d",&n,&L,&U); for(i=1;i<n;i++) { scanf("%d%d%lf",&x,&y,&z); tj(x,y,z); tj(y,x,z); if(lim<z) lim=z;//二分的右边界 } for(i=1;i<=n;i++) mx[i]=a[i]=-INF; sum=n; getroot(1,0); work(root); printf("%.3lf",ans); return 0; }