比较水的一道树上模拟水题,更新每个点的价值为以这个点为根的子树中的价值最小值,同时更新以每个节点为根的$sum$值,即以这个节点为根的子树的礼物数,
如果$sum$值小于自身所规定数,就用最小值来计算剩余价值。
#include<iostream> #include<cstdio> #include<algorithm> #define N 1000000 #define inf 0x7fffffff #define LL long long using namespace std; LL n,val[N],d[N],f[N],s[N]; LL ans; LL head[N],tot; struct node{ LL to,next; }e[N]; void add(LL u,LL v){ e[++tot].to=v,e[tot].next=head[u],head[u]=tot; } void dfs(LL u){ for(LL i=head[u];i;i=e[i].next){ LL v=e[i].to; if(v==f[u]) continue; dfs(v); val[u]=min(val[u],val[v]); s[u]+=s[v]; } if(s[u]<d[u]) ans+=(d[u]-s[u])*val[u],s[u]=d[u]; } int main() { scanf("%lld",&n); for(LL i=1;i<=n;i++){ scanf("%lld%lld%lld",&f[i],&d[i],&val[i]); add(i,f[i]),add(f[i],i); } for(LL i=1;i<=n;i++){ if(f[i]==-1){ dfs(i); printf("%lld\n",ans); break; } } return 0; }
洛谷——P3018 [USACO11MAR]树装饰Tree Decoration
原文:https://www.cnblogs.com/song-/p/9784203.html