标签:pre Gathering edg int USACO10MAR cnt maxn P2986 sum
一道转移类型的树形dp
考虑u点作为集合地点,不方便度为f,如果换做v点(v为u的一个儿子),不方便度会怎么变化
pre[u]表示在u点上方的点值之和,sum[u]表示u点子树点值之和
如果换做v点
f+(pre[u]+sum[u]-sum[v])w-sum[v]w
pre[u]+sum[u]-sum[v]表示这个边一头所有的点值之和
sum[v]表示这个边另一头所有的点值之和
所以问题变为了维护sum数组和pre数组
点击查看代码
#include<bits/stdc++.h>
using namespace std;
#define lowbit(x) x&(-x)
#define ll long long
const int maxn=1e5+5;
ll ans;
int cnt,head[maxn];
struct node{
int to,next,w;
}edg[maxn];
void add(int u,int v,int w){
edg[++cnt].next=head[u];head[u]=cnt;edg[cnt].to=v;edg[cnt].w=w;
}
int n;
ll val[maxn],dep[maxn],sum[maxn],pre[maxn];
void dfs1(int u,int fa);
void dfs2(int u,int fa,ll f);
int main(){
cin>>n;
for(int i=1;i<=n;i++)scanf("%d",&val[i]);
for(int aa,bb,ww,i=2;i<=n;i++){
scanf("%d%d%d",&aa,&bb,&ww);
add(aa,bb,ww);add(bb,aa,ww);
}
dfs1(1,0);
for(int i=2;i<=n;i++)
ans+=val[i]*dep[i];
dfs2(1,1,ans);
printf("%lld\n",ans);
return 0;
}
void dfs1(int u,int fa){
sum[u]=val[u];
for(int i=head[u];i;i=edg[i].next){
int to=edg[i].to,w=edg[i].w;
if(to==fa)continue;
dep[to]=dep[u]+w;
dfs1(to,u);
sum[u]+=sum[to];
}
}
void dfs2(int u,int fa,ll f){
pre[u]=pre[fa]+sum[fa]-sum[u];
for(int i=head[u];i;i=edg[i].next){
int to=edg[i].to,w=edg[i].w;
if(to==fa)continue;
ans=min(ans,f+(pre[u]+sum[u]-sum[to])*w-sum[to]*w);
dfs2(to,u,f+(pre[u]+sum[u]-sum[to])*w-sum[to]*w);
}
}
标签:pre,Gathering,edg,int,USACO10MAR,cnt,maxn,P2986,sum 来源: https://www.cnblogs.com/wzxbeliever/p/16067325.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。