1. 程式人生 > 其它 >P6847-[CEOI2019]Magic Tree【dp,線段樹合併】

P6847-[CEOI2019]Magic Tree【dp,線段樹合併】

正題

題目連結: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;
}