P6847-[CEOI2019]Magic Tree【dp,線段樹合併】
阿新 • • 發佈:2021-07-21
正題
題目連結:https://www.luogu.com.cn/problem/P6847
題目大意
\(n\)個點的一棵樹上,每個時刻可以割掉一些邊,一些節點上有果實表示如果在\(d_i\)時刻這個點恰好不與\(1\)聯通,那麼就可以獲得\(w_i\)的價值。
\(1\leq n,k\leq 10^5\)
解題思路
設\(f_{x,i}\)表示節點\(x\)在時刻\(i\)之前割掉時的最大權值那麼相當與在兒子裡面選一個最大的\(f_{y,j}(j\leq i)\)合併上來。
這是一個很經典的轉移方式,和命運那題一樣,直接用線段樹合併維護就好了。
時間複雜度\(O(n\log k)\)
code
#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=1e5+10; ll n,m,k,rt[N],fa[N],d[N],w[N]; ll cnt,t[N<<5],lazy[N<<5],ls[N<<5],rs[N<<5]; void Downdata(int x){ if(!lazy[x])return; if(ls[x])lazy[ls[x]]+=lazy[x],t[ls[x]]+=lazy[x]; if(rs[x])lazy[rs[x]]+=lazy[x],t[rs[x]]+=lazy[x]; lazy[x]=0;return; } void Change(ll &x,ll L,ll R,ll pos,ll val,ll z){ if(!x)x=++cnt; if(L==R){t[x]=val+max(z,t[x]);return;} ll mid=(L+R)>>1;Downdata(x); if(pos<=mid)Change(ls[x],L,mid,pos,val,z); else Change(rs[x],mid+1,R,pos,val,max(z,t[ls[x]])); t[x]=max(t[ls[x]],t[rs[x]]); return; } ll Merge(ll L,ll R,ll x,ll y,ll mx1,ll mx2){ if(!x||!y){ if(x)lazy[x]+=mx2,t[x]+=mx2; if(y)lazy[y]+=mx1,t[y]+=mx1; return x|y; } if(L==R){t[x]=max(t[x],mx1)+max(t[y],mx2);return x;} ll mid=(L+R)>>1;Downdata(x);Downdata(y); rs[x]=Merge(mid+1,R,rs[x],rs[y],max(mx1,t[ls[x]]),max(mx2,t[ls[y]])); ls[x]=Merge(L,mid,ls[x],ls[y],mx1,mx2); t[x]=max(t[ls[x]],t[rs[x]]); return x; } signed main() { scanf("%lld%lld%lld",&n,&m,&k); for(ll i=2;i<=n;i++) scanf("%lld",&fa[i]); for(ll i=1;i<=m;i++){ ll x;scanf("%lld",&x); scanf("%lld%lld",&d[x],&w[x]); } for(ll x=n;x>=1;x--){ if(d[x])Change(rt[x],1,k,d[x],w[x],0); if(fa[x])rt[fa[x]]=Merge(1,k,rt[fa[x]],rt[x],0,0); } printf("%lld\n",t[rt[1]]); return 0; }