[Luogu1552][APIO2012]派遣--(左偏樹)
阿新 • • 發佈:2018-05-16
最大 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]派遣--(左偏樹)