二分答案$mid$,若存在一条路径满足$|ave-k|<mid$,则答案至多为$mid-1$。
若$ave\leq k$,则$\sum(w-k)\leq 0$,且$\sum(k-w-mid)<0$;若$ave\geq k$,那么同理。
预先树分治处理出所有到重心的路径的信息,并按$w-k$排序。
那么每次检查的时候,通过双指针满足第一个限制,维护最小值来满足第二个限制。
需要注意的是,不能选取两条来自同一棵子树的路径,这只需要维护来自不同子树的最小的两条路径即可。
时间复杂度$O(n\log n(\log n+\log w))$。
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=50010,M=1000010;
int n,i,x,y,g[N],nxt[N<<1],v[N<<1],ok[N<<1],ed,son[N],f[N],all,now;
ll z,K,w[N<<1],val[M],L,R,mid;
int st[N],en[N],cnt,cur;
struct P{ll x;int y,z;P(){}P(ll _x,int _y,int _z){x=_x,y=_y,z=_z;}}q[2][M];
inline bool cmp(const P&a,const P&b){return a.x<b.x;}
inline void add(int x,int y,ll z){v[++ed]=y;w[ed]=z;nxt[ed]=g[x];g[x]=ed;ok[ed]=1;}
void findroot(int x,int y){
son[x]=1;f[x]=0;
for(int i=g[x];i;i=nxt[i])if(ok[i]&&v[i]!=y){
findroot(v[i],x);
son[x]+=son[v[i]];
if(son[v[i]]>f[x])f[x]=son[v[i]];
}
if(all-son[x]>f[x])f[x]=all-son[x];
if(f[x]<f[now])now=x;
}
void dfs(int x,int y,ll a,int b,int c){
q[0][++cur]=P(a,b,c);
q[1][cur]=P(-a,b,c);
for(int i=g[x];i;i=nxt[i])if(ok[i]&&v[i]!=y)dfs(v[i],x,a+w[i]-K,b+1,c);
}
void solve(int x){
int i,o=++cnt;
st[o]=cur+1;
for(i=g[x];i;i=nxt[i])if(ok[i])dfs(v[i],x,w[i]-K,1,v[i]);
en[o]=cur;
for(i=0;i<2;i++)sort(q[i]+st[o],q[i]+en[o]+1,cmp);
for(i=g[x];i;i=nxt[i])if(ok[i]){
ok[i^1]=0;
f[0]=all=son[v[i]];
findroot(v[i],now=0);
solve(now);
}
}
inline bool check(){
for(int i=0;i<2;i++){
for(int j=1;j<=cur;j++){
val[j]=-q[i][j].x-mid*q[i][j].y;
if(q[i][j].x<=0&&val[j]<0)return 1;
}
for(int j=1;j<=cnt;j++){
int l=st[j],r=en[j],x=r,y=l,az=0,bz=0;ll av,bv;
for(;x>=l;x--){
for(;y<=r&&q[i][x].x+q[i][y].x<=0;y++){
ll v=val[y];int z=q[i][y].z;
if(az==z){if(av>v)av=v;}
else if(!az||av>v)bv=av,bz=az,av=v,az=z;
else if(!bz||bv>v)bv=v,bz=z;
}
if(az&&az!=q[i][x].z&&val[x]+av<0)return 1;
if(bz&&bz!=q[i][x].z&&val[x]+bv<0)return 1;
}
}
}
return 0;
}
int main(){
scanf("%d%lld",&n,&K);
for(ed=i=1;i<n;i++)scanf("%d%d%lld",&x,&y,&z),add(x,y,z),add(y,x,z);
f[0]=all=n;findroot(1,now=0);solve(now);
L=1,R=1e13;
while(L<=R){
mid=(L+R)>>1;
if(check())R=mid-1;else L=mid+1;
}
return printf("%lld",R),0;
}