[noi2011]道路修建 树形dp

时间:2023-03-08 23:19:31
[noi2011]道路修建 树形dp

这道题可以说是树形dp的入门题,也可以看成是一道检验【树】这个数据结构的题目;

这道题只能bfs,毕竟10^6的复杂度win下肯定爆栈了;

但是最恶心的还不是这个,实测用printf输出

[noi2011]道路修建 树形dp

用cout输出

[noi2011]道路修建 树形dp

题上也不提醒一下,无语啦;

 #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<string>
#include<iomanip>
#include<cmath>
using namespace std;
#define LL long long
const int maxn=;
int n;
struct node{
int y,next,v;
}e[maxn<<];
int linkk[maxn],len=,siz[maxn],q[maxn],tail=,fa[maxn],t[maxn],top=,ru[maxn],w[maxn];
LL ans=;
void insert(int x,int y,int v){
e[++len].y=y;
e[len].next=linkk[x];
e[len].v=v;
linkk[x]=len;
}
void init(){
scanf("%d",&n);
int x,y,v;
for(int i=;i<n;i++){
scanf("%d%d%d",&x,&y,&v);
insert(x,y,v);insert(y,x,v);
}
}
void bfs(){
q[++tail]=;int x;
for(int head=;head<=tail;head++){
x=q[head];
for(int i=linkk[x];i;i=e[i].next){
if(e[i].y==fa[x])continue;
q[++tail]=e[i].y;
fa[e[i].y]=x;
ru[x]++;
w[e[i].y]=e[i].v;
}
if(!ru[x])t[++top]=x;
}
for(int head=;head<=top;head++){
x=t[head];siz[x]++;
siz[fa[x]]+=siz[x];
ru[fa[x]]--;
if(!ru[fa[x]])t[++top]=fa[x];
ans+=(LL)abs(n-*siz[x])*(LL)w[x];
}
}
void work(){
bfs();
cout<<ans<<endl;
}
int main(){
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
init();
work();
}