洛谷P1364医院设置

洛谷P1364 医院设置

医院设置

题目描述

设有一棵二叉树,如图:

其中,圈中的数字表示结点中居民的人口。圈边上数字表示结点编号,现在要求在某个结点上建立一个医院,使所有居民所走的路程之和为最小,同时约定,相邻接点之间的距离为 1 1 1。如上图中,若医院建在 1 1 1 处,则距离和 = 4 + 12 + 2 × 20 + 2 × 40 = 136 =4+12+2\times20+2\times40=136 =4+12+2×20+2×40=136;若医院建在 3 3 3 处,则距离和 = 4 × 2 + 13 + 20 + 40 = 81 =4\times2+13+20+40=81 =4×2+13+20+40=81。

输入格式

第一行一个整数 n n n,表示树的结点数。

接下来的 n n n 行每行描述了一个结点的状况,包含三个整数 w , u , v w, u, v w,u,v,其中 w w w 为居民人口数, u u u 为左链接(为 0 0 0 表示无链接), v v v 为右链接(为 0 0 0 表示无链接)。

输出格式

一个整数,表示最小距离和。

样例 #1

样例输入 #1

复制代码
5						
13 2 3
4 0 0
12 4 5
20 0 0
40 0 0

样例输出 #1

复制代码
81

提示

数据规模与约定

对于 100 % 100\% 100% 的数据,保证 1 ≤ n ≤ 100 1 \leq n \leq 100 1≤n≤100, 0 ≤ u , v ≤ n 0 \leq u, v \leq n 0≤u,v≤n, 1 ≤ w ≤ 1 0 5 1 \leq w \leq 10^5 1≤w≤105。

正确代码

cpp 复制代码
//
// Created by 枫飘长安 on 2024/4/15.
//
#include <bits/stdc++.h>
using namespace std;
/**
  *author: 枫飘长安
  *date: 2024/4/15
  *fileName: P1364.cpp
  *description: TODO
  **/
typedef long long ll;
const int MAXN = 1e6+10;

int ans = 0x3f3f3f3f;
int n;
bool graph[105][105];
bool vis[105];
int num[105];


struct node {
    int u,step;
};

int bfs(int x) {
    memset(vis,0,sizeof(vis));
    queue<node> q;
    vis[x] = true;
    q.push((node){x,0});
    int sum = 0;
    while(!q.empty()){
        node temp=q.front();
        q.pop();
        for(int i=1;i<=n;i++)
            if(graph[temp.u][i]&&!vis[i]){
                vis[i] = true;
                q.push({i,temp.step+1});
                sum+=num[i]*(temp.step+1);
            }
    }
    return sum;
}

int main (){
    cin >> n;
    for (int i = 1; i <= n; i++) {
        int l,r;
        cin>> num[i] >> l >> r;
        if (l != 0) {
            graph[i][l] = true;
            graph[l][i] = true;
        }
        if (r != 0) {
            graph[i][r] = true;
            graph[r][i] = true;
        }
    }
    for (int i = 1; i <=n; i++) {
        ans = min(ans,bfs(i));
    }
    cout << ans;
    return 0;
}
相关推荐
程序员莫小特1 天前
老题新解|大整数加法
数据结构·c++·算法
过往入尘土1 天前
服务端与客户端的简单链接
人工智能·python·算法·pycharm·大模型
zycoder.1 天前
力扣面试经典150题day1第一题(lc88),第二题(lc27)
算法·leetcode·面试
蒙奇D索大1 天前
【数据结构】考研数据结构核心考点:二叉排序树(BST)全方位详解与代码实现
数据结构·笔记·学习·考研·算法·改行学it
智驱力人工智能1 天前
工厂抽烟检测系统 智能化安全管控新方案 加油站吸烟检测技术 吸烟行为智能监测
人工智能·算法·安全·边缘计算·抽烟检测算法·工厂抽烟检测系统·吸烟监测
程序员爱钓鱼1 天前
Go语言实战案例——进阶与部署篇:编写Makefile自动构建Go项目
后端·算法·go
_Power_Y1 天前
Java面试常用算法api速刷
java·算法·面试
艾醒(AiXing-w)1 天前
大模型面试题剖析:模型微调中冷启动与热启动的概念、阶段与实例解析
人工智能·深度学习·算法·语言模型·自然语言处理
天选之女wow1 天前
【代码随想录算法训练营——Day32】动态规划——509.斐波那契数、70.爬楼梯、746.使用最小花费爬楼梯
算法·leetcode·动态规划
红衣小蛇妖1 天前
LeetCode-704-二分查找
java·算法·leetcode·职场和发展