点此看题
首先选子树为 \(dp\) 主体,但是考虑没有时间做不动,我们假设子树 \(u\) 是在时间 \(i\) 被断开的,也就是断开操作是由于 \(u\) 的祖先引起的。设 \(dp[u][i]\) 表示子树 \(u\) 在 \(i\) 时间内被断开最后能得到的最大果汁,转移讨论 \(u\) 这个点的果汁:
显然可以用线段树合并来优化,第一种转移就直接线段树合并,第二种转移不能维护区间最值标记,因为这样就不能合并。因为 \(dp[u][i]\) 关于 \(i\) 单调所以第二种转移可以考虑成区间赋值。
具体实现中区间赋值不打标记,而是线段树上的点维护 \(\min,\max\),如果 \(\min=\max\) 的时候就说明这个区间是一个值,线段树合并的时候如果遇到区间同值的情况就打加法标记,修改的时候如果区间同值就新开儿子节点,上传的时候如果发现区间同值可以把儿子节点删掉(方便理解,避免出锅)
那么时间复杂度 \(O(n\log n)\),警告:一定不要尝试去维护区间赋值标记,很难。
还有一种小清新的优化方法,因为 \(dp\) 值单调所以可以考虑差分,那么第一种转移就能直接启发式合并,第二种转移是增加 \(w_u\) 的差分值,直接插入 \(\tt set\) 中,这其实是取 \(\max\) 操作,所以要从后面删除一些差分标记,时间复杂度 \(O(n\log^2n)\),但是比线段树合并要少 \(100\) 多行。
\(\tt think\ twice,code\ once\),后期发现一个思路是否错误的能力真的很重要。
当 \(dp\) 值单调且需要合并的时候可以考虑维护差分值。线段树合并是加法标记是最支持的,其他好像都不怎么好。
#include <cstdio> #include <iostream> using namespace std; const int M = 100005; const int N = 100*M; #define ll long long int read() { int x=0,f=1;char c; while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;} while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();} return x*f; } int n,m,k,tot,f[M],d[M],w[M],rt[M]; int cnt,ls[N],rs[N];ll mi[N],ad[N],mx[N]; struct edge { int v,next; }e[M]; int same(int x) { return !ls[x] && !rs[x]; } void add(int x,ll y) { if(!x) return ; mi[x]+=y;mx[x]+=y;ad[x]+=y; } void up(int x) { mx[x]=max(mx[ls[x]],mx[rs[x]]); mi[x]=min(mi[ls[x]],mi[rs[x]]); if(mx[x]==mi[x]) ls[x]=rs[x]=0;//delete } void down(int x) { if(ad[x]) { add(ls[x],ad[x]); add(rs[x],ad[x]); ad[x]=0; } } int merge(int x,int y) { if(!x || !y) return x+y; if(same(y)) { add(x,mx[y]); return x; } if(same(x)) { add(y,mx[x]); return y; } down(x);down(y); ls[x]=merge(ls[x],ls[y]); rs[x]=merge(rs[x],rs[y]); up(x); return x; } ll ask(int x,int l,int r,int p) { if(same(x)) return mx[x]; int mid=(l+r)>>1;down(x); if(mid>=p) return ask(ls[x],l,mid,p); return ask(rs[x],mid+1,r,p); } void upd(int &x,int l,int r,int L,int R,ll v) { if(l>R || L>r || mi[x]>=v) return ; if(mx[x]<=v && L<=l && r<=R) { mx[x]=mi[x]=v; ls[x]=rs[x]=ad[x]=0; return ; } int mid=(l+r)>>1;down(x); if(same(x)) { ls[x]=++cnt;rs[x]=++cnt; mx[ls[x]]=mi[ls[x]]=mx[rs[x]]=mi[rs[x]]=mx[x]; } upd(ls[x],l,mid,L,R,v); upd(rs[x],mid+1,r,L,R,v); up(x); } void dfs(int u) { rt[u]=++cnt; for(int i=f[u];i;i=e[i].next) { int v=e[i].v; dfs(v); rt[u]=merge(rt[u],rt[v]); } if(d[u]) upd(rt[u],1,k,d[u],k,w[u]+ask(rt[u],1,k,d[u])); } signed main() { n=read();m=read();k=read(); for(int u=2;u<=n;u++) { int v=read(); e[++tot]=edge{u,f[v]},f[v]=tot; } for(int i=1;i<=m;i++) { int v=read(); d[v]=read();w[v]=read(); } dfs(1); printf("%lld\n",mx[rt[1]]); }