LUOGU P2052 [NOI2011]道路修建

最新推荐文章于 2024-03-11 20:47:49 发布

转载最新推荐文章于 2024-03-11 20:47:49 发布 · 163 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/sdfzsyq/p/9676843.html

本文解析了一道被认为是NOI比赛中较为复杂的问题，并通过深入浅出的方式解释了解题思路。文章强调理解题目真正含义的重要性，并提供了一种利用深度优先搜索（DFS）遍历树形结构来高效求解的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门

解题思路

这题怕是noi的耻辱，理解题意有点困难，是建好树以后算，并不是决策。理解题意后就十分简单了，一遍dfs。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#define LL long long

using namespace std;
const int MAXN=1e6+5;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?0:1;ch=getchar();}
    while(isdigit(ch))  {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return f?x:-x;
} 

int n,head[MAXN],cnt,to[MAXN<<1],nxt[MAXN<<1],val[MAXN<<1];
int siz[MAXN];
LL ans;

inline void add(int bg,int ed,int w){
    to[++cnt]=ed,nxt[cnt]=head[bg],val[cnt]=w,head[bg]=cnt;
}

void dfs(int x,int pre){
    siz[x]=1;
    for(register int i=head[x];i;i=nxt[i]){
        int u=to[i];if(u==pre) continue;
        dfs(u,x);
        siz[x]+=siz[u];
        ans+=(LL)val[i]*abs(n-siz[u]-siz[u]);
    }
}

int main(){
    n=rd();int x,y,z;
    for(register int i=1;i<n;i++){
        x=rd(),y=rd(),z=rd();
        add(x,y,z),add(y,x,z);
    }
    dfs(1,0);cout<<ans<<endl;
    return 0;
}