UOJ #276. 【清华集训2016】汽水

TMD写了一篇博客竟然还不够清醒,那就再写一篇睡觉去了

首先一看题目就会发现这可以先套上个分数规划,即我们现在要最小化\(|\frac{\sum_{i=1}^{len} w_i}{len}-k|\)

考虑二分答案\(x\),顺便拆掉绝对值,即\(-x\le\frac{\sum_{i=1}^{len} w_i}{len}-k\le x\),发现其实只要同时满足:

\[\sum_{i=1}^{len} w_i-k+x \ge 0\]

\[\sum_{i=1}^{len} w_i-k-x \le 0\]

所以\(k\)不用管直接减掉即可,考虑就是找一条路径同时满足上述两个条件

然后又考虑到这是树上问题,我们考虑点分治,如果单看第一个条件很简单,只要把分治中心到所有点的边权和搞出来,找两条(或一条)长度大于\(0\)的即可

那么还有第二个要求怎么办,不难发现因为只用取两条链,因此我们可以在满足第一个的情况下最小化第二个的值,这个可以排序后two points扫一下解决

注意一下两条链都在同一子树内的情况要判一下,总体复杂度\(O(n\log^2 n)\)

#include<cstdio>
#include<iostream>
#include<algorithm>
#define int long long
#define RI register int
#define CI const int&
using namespace std;
typedef long double DB;
const int N=100005,INF=1e18;
const DB EPS=1e-3;
struct edge
{
    int to,nxt; DB v1,v2;
}e[N<<1]; int n,head[N],cnt,k,x,y,z;
inline void addedge(CI x,CI y,CI z)
{
    e[++cnt]=(edge){y,head[x],(DB)z-k,(DB)z-k}; head[x]=cnt;
    e[++cnt]=(edge){x,head[y],(DB)z-k,(DB)z-k}; head[y]=cnt;
}
class Point_Division_Solver
{
    private:
        struct data
        {
            int id; DB d1,d2;
            friend inline bool operator < (const data& A,const data& B)
            {
                return A.d1<B.d1;
            }
        }d[N]; int size[N],mx[N],ots,tot,rt; bool vis[N];
        #define to e[i].to
        inline void getrt(CI now,CI fa=0)
        {
            size[now]=1; mx[now]=0; for (RI i=head[now];i;i=e[i].nxt) if (to!=fa&&!vis[to])
            getrt(to,now),size[now]+=size[to],mx[now]=max(mx[now],size[to]);
            if ((mx[now]=max(mx[now],ots-size[now]))<mx[rt]) rt=now;
        }
        inline void travel(CI now,CI fr,CI fa,const DB& dis1,const DB& dis2)
        {
            d[++tot]=(data){fr,dis1,dis2}; for (RI i=head[now];i;i=e[i].nxt)
            if (to!=fa&&!vis[to]) travel(to,fr,now,dis1+e[i].v1,dis2+e[i].v2);
        }
        inline void insert(int& mi,int& smi,CI p)
        {
            if (d[p].d2<d[mi].d2) { if (d[p].id!=d[mi].id) smi=mi; mi=p; }
            else if (d[p].d2<d[smi].d2&&d[p].id!=d[mi].id) smi=p;
        }
        inline bool judge(CI rt)
        {
            RI i,pos,mi,smi; for (d[tot=1]=(data){rt,0,0},i=head[rt];i;i=e[i].nxt)
            if (!vis[to]) travel(to,to,rt,e[i].v1,e[i].v2); 
            sort(d+1,d+tot+1); d[0].d2=INF; for (i=pos=1;d[i].d1<0;++i,++pos);
            for (mi=smi=0;i<=tot;++i)
            {
                while (pos>1&&d[pos-1].d1+d[i].d1>=0) insert(mi,smi,--pos);
                if (d[mi].id!=d[i].id&&d[mi].d2+d[i].d2<=0) return 1;
                if (d[mi].id==d[i].id&&d[smi].d2+d[i].d2<=0) return 1;
                insert(mi,smi,i);
            }
            return 0;
        }
        inline bool solve(CI now)
        {
            vis[now]=1; if (judge(now)) return 1;
            for (RI i=head[now];i;i=e[i].nxt) if (!vis[to])
            {
                mx[rt=0]=INF; ots=size[to]; getrt(to);
                if (solve(rt)) return 1;
            }
            return 0;
        }
        #undef to
    public:
        inline bool check(const DB& x)
        {
            //printf("%lld : ",(int)x); 
            RI i; for (i=1;i<=n;++i) vis[i]=0;
            for (i=1;i<=cnt;++i) e[i].v1+=x,e[i].v2-=x;
            mx[rt=0]=INF; ots=n; getrt(1); bool flag=solve(rt);
            for (i=1;i<=cnt;++i) e[i].v1-=x,e[i].v2+=x; return flag;
        }
}PD;
signed main()
{
    //freopen("CODE.in","r",stdin); freopen("CODE.out","w",stdout);
    RI i; for (scanf("%lld%lld",&n,&k),i=1;i<n;++i)
    scanf("%lld%lld%lld",&x,&y,&z),addedge(x,y,z);
    DB l=0,r=1e13,mid; while (r-l>EPS)
    if (PD.check(mid=(l+r)/2.0)) r=mid; else l=mid;
    return printf("%lld",(int)(r)),0;
}

猜你喜欢

转载自www.cnblogs.com/cjjsb/p/11616519.html