1. 程式人生 > >[Luogu1552][APIO2012]派遣--(左偏樹)

[Luogu1552][APIO2012]派遣--(左偏樹)

最大 cost 題目 define 個數 top apio2012 ios HR

首先,洛谷傳送門

題目大意:

有一顆n個節點的樹,每個節點都有li(領導值),fi(父親),ci(cost)三個值,要求選定一個節點x,並在其與其子樹中選出任意數量的點,使所有選的點的總花費x不超過m,使權值(所選點的個數*lx)最大。

解題思路:

其實這題只要看懂題了還是很好做的,首先很好想到樹形DP,對於每個節點,可以選出在其子樹中最小的幾個節點使其花費剛好不超過m,在乘上自己的l值,但每次去取最小花費的節點時間復雜度過於復雜,因此,可以想到用可並堆來維護每個節點的兒子,再向父親轉移即可,因此最壞復雜度為(nlogn)。

代碼:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define For(i,j,k) for(int i=j;i<=k;++i)
#define Dor(i,j,k) for(int i=j;i>=k;--i)
#define Ror(i,x) for(int i=head[x];i;i=e[i].nxt)
#define fin(s) freopen(s".in","r",stdin)
#define fout(s) freopen(s".out","w",stdout)
#define ls(x) (ch[x][0])
#define rs(x) (ch[x][1])
#define ll long long
using namespace std;
const int N=1e5+10;
struct zps{
    int ch[N][2],dis[N],f[N];ll w[N];
    int merge(int x,int y){
        if(!x||!y)return x+y;
        if(w[x]<w[y])swap(x,y);
        rs(x)=merge(rs(x),y);
        f[rs(x)]=x;
        if(dis[rs(x)]>dis[ls(x)])swap(ls(x),rs(x));
        dis[x]=dis[rs(x)]+1;
        return x;
    }
    inline int del(int x){
        f[ls(x)]=f[rs(x)]=0;
        int ret=merge(ls(x),rs(x));
        ls(x)=rs(x)=0;
        return ret;
    }
    int find(int x){return !f[x]?x:find(f[x]);}
}T;
struct edge{
    int to,nxt;
}e[N<<1];
int head[N],tot,n;
inline void add(int x,int y){
    e[++tot]=(edge){y,head[x]},head[x]=tot;
    e[++tot]=(edge){x,head[y]},head[y]=tot;
}
ll l[N],a[N],ans,m,siz[N];int top[N],rt;
inline void dfs(int x,int z){
    a[x]=T.w[x];siz[x]=1;top[x]=x;
    Ror(i,x){
        int to=e[i].to;
        if(to==z)continue;
        dfs(to,x);
        if(!top[to])continue;
        top[x]=T.merge(top[x],top[to]);
        a[x]+=a[to];
        siz[x]+=siz[to];
    }
    while(a[x]>m){
        a[x]-=T.w[top[x]];
        int tt=T.del(top[x]);
        top[x]=tt;
        --siz[x];
    }
    ans=max(ans,siz[x]*l[x]);
}
int main(){
//  fin("p1552");
//  fout("p1552");
    scanf("%d%lld",&n,&m);
    int x;
    For(i,1,n){
        scanf("%d%lld%lld",&x,&T.w[i],&l[i]);
        if(!x)rt=i;
        else add(x,i);
    }
    dfs(rt,rt);
    printf("%lld\n",ans);
    return 0;
}

總結:

其實這題作為左偏樹的入門題對於新手來說很好,其中有許多小點需要考慮到,需要細心。

最後...

Think twice,code once.

[Luogu1552][APIO2012]派遣--(左偏樹)