UOJ #276「清华集训2016」汽水

为什么你们常数都这么小啊

UOJ #276

题意：在树上找一条链使得|边权平均值$ -k$|尽量小,$ n<=5e4$

---

$ Solution:$

首先二分答案$ ans$，即我们需要找一条链使得边权平均值 $\in [-ans,ans]$

我们分正负两半分开讨论

先假设平均值$ \in (0,ans]$

将原树点分

统计过根的所有链

将这些链记录长度$len$,边数$sum$,所属子树标号$id$之后按长度排序

添加一条链$(0,0,0)$，则过某点的链一定是某两条不在同一子树的链拼接而成

两条链拼接合法当且仅当$ \frac{len_i+len_j}{sum_i+sum_j} \in (0,ans]$

化简得$ len_i+len_j>0且len_i-ans·sum_i+len_j-ans·sum_j \leq 0$

我们记录后缀$ len_i-ans·sum_i$的最小值

然后两个指针扫扫就好了

注意存在不在同一子树的限制，我们需要同时记录不在同一子树的后缀最小值和次小值，这样才能合并

如果平均值$ \in [-ans,0]$，也用同样的方法维护前缀最大值和次大值即可

注意各种边界的问题

时间复杂度:$ O(n \ log \ n \ log \ val)$

---

$ my \ code$

#include<ctime>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#define M 100010
#define rt register int
#define ll long long
using namespace std;
inline ll read(){
    ll x = ; char zf = ; char ch = getchar();
    while (ch != '-' && !isdigit(ch)) ch = getchar();
    if (ch == '-') zf = -, ch = getchar();
    while (isdigit(ch)) x = x *  + ch - '', ch = getchar(); return x * zf;
}
void write(ll y){if(y<)putchar('-'),y=-y;if(y>)write(y/);putchar(y%+);}
void writeln(const ll y){write(y);putchar('\n');}
int i,j,k,m,n,x,y,z,cnt,all,Root,nowmin;
int F[M],L[M],N[M],a[M],size[M];ll c[M];
bool vis[M];
void add(int x,int y,ll z){
    a[++k]=y;c[k]=z;
    if(!F[x])F[x]=k;
    else N[L[x]]=k;
    L[x]=k;
}
void getRoot(int x,int pre){
    size[x]=;int Maxsize=;
    for(rt i=F[x];i;i=N[i])if(a[i]!=pre&&!vis[a[i]]){
        getRoot(a[i],x);size[x]+=size[a[i]];
        Maxsize=max(Maxsize,size[a[i]]);
    }
    Maxsize=max(Maxsize,all-size[x]);
    if(Maxsize<nowmin)nowmin=Maxsize,Root=x;
}
int cs;
vector<int>e[];
void build(int x){
    int ls=all;vis[x]=;
    for(rt i=F[x];i;i=N[i])if(!vis[a[i]]){
        all=(size[a[i]]>size[x]?ls-size[x]:size[a[i]]);
        nowmin=;Root=;getRoot(a[i],x);
        e[x].push_back(Root);
        build(Root);
    }
}
struct chain{
    ll len;int sl,col;
    bool operator <(const chain s)const{
        return len<s.len;
    }
}q[];int t;
void Add(int col,int x,int pre,ll len,int sl){
    q[++t]={len,sl,col};
    for(rt i=F[x];i;i=N[i])if(a[i]!=pre&&!vis[a[i]])Add(col,a[i],x,len+c[i],sl+);
}
int qzmax[][],hzmin[][],allsize=;
bool check(int x,ll ans){//0...x
    q[t=]={,,};vis[x]=;
    for(rt i=F[x];i;i=N[i])if(!vis[a[i]])Add(a[i],a[i],x,c[i],);
    sort(q+,q+t+);
    qzmax[][]=;qzmax[][]=;
    for(rt i=;i<=t;i++){
        int id0=qzmax[i-][],id1=qzmax[i-][];
        qzmax[i][]=id0;qzmax[i][]=id1;
        if(q[i].len+ans*q[i].sl>q[id0].len+ans*q[id0].sl){
            if(q[i].col!=q[id0].col)qzmax[i][]=qzmax[i][];
            qzmax[i][]=i;
        }
        else if(q[i].len+ans*q[i].sl>q[id1].len+ans*q[id1].sl||!id1){
            if(q[i].col!=q[id0].col)qzmax[i][]=i;
        }
    }
    hzmin[t][]=t;hzmin[t][]=;
    for(rt i=t-;i>=;i--){
        int id0=hzmin[i+][],id1=hzmin[i+][];
        hzmin[i][]=id0;hzmin[i][]=id1;
        if(q[i].len-ans*q[i].sl<q[id0].len-ans*q[id0].sl){
            if(q[i].col!=q[id0].col)hzmin[i][]=hzmin[i][];
            hzmin[i][]=i;
        }
        else if(q[i].len-ans*q[i].sl<q[id1].len-ans*q[id1].sl||!id1){
            if(q[i].col!=q[id0].col)hzmin[i][]=i;
        }
    }
    int R=t;
    for(rt i=;i<=t;i++){
        while((q[i].len+q[R].len>||R==i)&&R)R--;
        if(R==)break;if(R>=i)continue;
        int maxv=qzmax[R][];
        if(q[qzmax[R][]].col==q[i].col)maxv=qzmax[R][];
        if(maxv==)continue;
        if(q[i].len+q[i].sl*ans+q[maxv].len+ans*q[maxv].sl>)return ;
    }
    int L=;
    for(rt i=t;i>=;i--){
        while((q[i].len+q[L].len<||L==i)&&L<=t)L++;
        if(L>t)break;if(L<=i)continue;
        int minv=hzmin[L][];
        if(q[hzmin[L][]].col==q[i].col)minv=hzmin[L][];
        if(!minv)continue;
        if(q[i].len-q[i].sl*ans+q[minv].len-ans*q[minv].sl<)return ;
    }
    bool res=;
    for(rt i=;i<e[x].size();i++)if(!res)res|=check(e[x][i],ans);
    return res;
}
int main(){
    n=read();ll w=read();
    for(rt i=;i<n;i++){
        x=read();y=read();ll z=read();
        add(x,y,z-w);
        add(y,x,z-w);
    }
    nowmin=;all=n;getRoot(,);
    int troot=Root;build(Root);
    memset(vis,,sizeof(vis));
    ll L=,R=;
    while(L<=R){
        memset(vis,,sizeof(vis));
        ll mid=(L+R)/;
        if(check(troot,mid))R=mid-;
        else L=mid+;
    }
    write(R);
    return ;
}