zoukankan      html  css  js  c++  java
  • BZOJ5252 八省联考2018林克卡特树(动态规划+wqs二分)

      假设已经linkcut完了树,答案显然是树的直径。那么考虑这条直径在原树中是怎样的。容易想到其是由原树中恰好k+1条点不相交的链(包括单个点)拼接而成的。因为这样的链显然可以通过linkcut拼接起来,而若选择不超过k条链则可能有链不得不被cut拆开,即使不会被拆开也可以通过选择单点来达到恰好k+1条(下设k=k+1)。

      那么问题变为在树上选择k条点不相交的链使边权和最大。最简单的dp就是设f[i][j]为i子树中选j条链的最大权值,且用一维012状态记录i这个点在子树中的度数,转移类似于一个树形背包。复杂度看起来是O(nk^2),实际上似乎是O(nk),可以拿60分。

      然而再往上即使复杂度去一个k也没有更高的分了。这暗示我们可以做到与k无关的复杂度。

      考虑去掉k的限制。这样仍然是一个dp,只是去掉了一维,复杂度变为O(n)。注意一些细节,比如说仅选择单点看做度数为1,按照210的顺序更新等等。

      k怎么办?考虑wqs二分,也叫凸优化、带权二分啥的。我们把选择x条链的代价设为x*cost。可以发现,如果把cost设成-inf,那么会选择n条链;如果设成inf,会选择0条链;而随着cost从-inf增加到inf,最优链数感觉上应该是单调不增的。那么我们可以二分cost,使得链数最终在k处停止,这个时候肯定求得了选k条链的最优方案,于是再补回cost就可以求得最优解了。至于这个东西是不是单调不增的呢?貌似是的。为什么呢?大胆猜想不用证明感觉一个东西便宜的时候都不买,涨价了更没有理由买啊。证明我也不懂。

      dp的时候记录一下最优的时候至少需要选择几条链。因为cost可能与一段链数对应,有可能无法刚好二分到选了k条链,不过这说明他们在该情况下都能作为最优解,则他们的加权(x*cost)答案相同。那么只要能够二分到这种加权答案就可以算出来k条链的答案了。于是保留选择的链数最接近k而又不超过k的答案,最终答案加上k*cost。

      还有一个细节,cost只需要在整数范围内二分就可以了。具体还是用坐标系来理解吧,链数对应的实际答案构成一个凸壳,用斜率为cost的直线去切这个凸壳,第一个碰上的点就是我们找到的答案。因为其中都是整点,用整数的斜率去切就可以了。但其实我也不知道我在说啥。

      “题目并不难。”

    #include<iostream> 
    #include<cstdio>
    #include<cmath>
    #include<cstdlib>
    #include<cstring>
    #include<algorithm>
    using namespace std;
    int read()
    {
        int x=0,f=1;char c=getchar();
        while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
        while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
        return x*f;
    }
    #define N 300010
    #define ll long long
    #define inf 300000000010
    #define max(a,b) (((a)>(b))?(a):(b))
    int n,k,p[N],t=0;
    ll ans;
    struct data2
    {
        ll x;int cnt;
        bool operator >(const data2&a) const
        {
            return x>a.x||x==a.x&&cnt<a.cnt;
        }
        data2 operator +(const data2&a) const
        {
            return (data2){x+a.x,cnt+a.cnt};
        }
        data2 operator -(const data2&a) const
        {
            return (data2){x-a.x,cnt-a.cnt};
        }
    }f[N][3];
    struct data{int to,nxt,len;
    }edge[N<<1];
    void addedge(int x,int y,int z){t++;edge[t].to=y,edge[t].nxt=p[x],edge[t].len=z,p[x]=t;}
    void dfs(int k,int from,ll cost)
    {
        f[k][0]=(data2){0,0};f[k][1]=(data2){-cost,1};f[k][2]=(data2){-inf,0};
        for (int i=p[k];i;i=edge[i].nxt)
        if (edge[i].to!=from)
        {
            int y=edge[i].to;
            dfs(y,k,cost);
            data2 w=max(f[y][0],max(f[y][1],f[y][2]));
            data2 u=f[y][1]+(data2){edge[i].len+cost,-1},v=f[y][0]+(data2){edge[i].len,0};
            f[k][2]=max(f[k][2]+w,f[k][1]+max(u,v));
            u=f[y][1]+(data2){edge[i].len,0},v=f[y][0]+(data2){edge[i].len-cost,1};
            f[k][1]=max(f[k][1]+w,f[k][0]+max(u,v));
            f[k][0]=f[k][0]+w;
        }
        
    }
    int main()
    {
    #ifndef ONLINE_JUDGE
        freopen("bzoj5252.in","r",stdin);
        freopen("bzoj5252.out","w",stdout);
        const char LL[]="%I64d";
    #else
        const char LL[]="%lld";
    #endif
        n=read(),k=read()+1;
        for (int i=1;i<n;i++)
        {
            int x=read(),y=read(),z=read();
            addedge(x,y,z),addedge(y,x,z);
        }
        ll l=-inf,r=inf;
        while (l<=r)
        {
            ll mid=l+r>>1;
            dfs(1,1,mid);
            if (max(f[1][0],max(f[1][1],f[1][2])).cnt<=k) ans=max(f[1][0],max(f[1][1],f[1][2])).x+k*mid,r=mid-1;
            else l=mid+1;
        }
        cout<<ans;
        return 0;
    }
  • 相关阅读:
    each和foreach的区别
    apply和call的用法
    js小知识点
    关于react的一些疑问点
    c语言字符动画的实现
    解决'chromedriver' executable needs to be in PATH问题!
    二叉树的创建和遍历
    dns和dhcp
    编写一个application程序实现如下功能:接受命令行中给出的一个字符串,先将字符串原样输出,然后判断该穿的第一个字母是否为大写,若是大写则统计该串中大写字母的个数,并将所有大写字母输出。
    学生成绩管理系统究极版
  • 原文地址:https://www.cnblogs.com/Gloid/p/9433783.html
Copyright © 2011-2022 走看看