【DFS】洛谷P1364 医院设置含代码详解

残夏丰秋源

已于 2024-07-21 16:27:05 修改

阅读量506

点赞数 21

分类专栏： DFS 文章标签：深度优先算法 c++ dfs

于 2024-07-21 16:25:15 首次发布

本文链接：https://blog.csdn.net/wayua/article/details/140589487

版权

DFS 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

洛谷医院设置 DFS

看看题👉洛谷P1364医院设置👇

医院设置

题目描述

设有一棵二叉树，如图：

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

样例输出 #1

提示

数据规模与约定

对于 $100\%$ 的数据，保证 $\leq n \leq 100$ ， $\leq u, v \leq n$ ， $\leq w \leq 10^5$ 。

非常经典的一道DFS题目，与其他DFS最大的不同点是需要建立双向边，并且开标记数组来标记，使已经走过的路不被再走一次。
这里的二叉树也比较特别，在不同的位置开始搜索（使每一个点都成为根节点），也就组成了不同的许多树（这些树最多可以成为三叉树）。
还有一点，就是每一个结点都有一个权值。

具体来说，就是：

读入n个结点，并且存上权值，让每一个结点成为根节点搜索一遍，组成许多不同的树。
每一次搜索时，用一个sum来累加每个权值与深度-1的积，最后得到一个sum，也就是距离之和。
得到每一次的sum，取最小的一个，并输出。

看一看代码的具体操作，里面有注释👇

#include<bits/stdc++.h>
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);//加快输入输出
#define pb push_back//定义push_back为pb
using namespace std;
const int N = 1e6;
long long arr[N],n,sum,minn = 1e18;
bool vis[N];//开一个标记数组
vector <int> v[N];//开vector数组存树

void dfs(int now,int dep)
{
	vis[now] = 1;//走到哪就把哪标记了
	sum += arr[now]*(dep-1);//sum累加
	for( auto t:v[now])//auto一个t来搜索
	{
		if(vis[t] == 0)		dfs(t,dep+1);//判断是否没被标记（即没走过），继续搜素
	}
}

int main()
{
	IOS;
	cin>>n;//读入结点数
	for(int i = 1;i <= n;i++)
	{
		int x,a,b;
		cin>>x>>a>>b;
		arr[i] = x;//将权值存到数组里
		if(a != 0)//判断有无子结
		{
			v[i].pb(a);//连接
			v[a].pb(i);//反向连接
		}
		if(b != 0)//判断有无子结
		{
			v[i].pb(b);//连接
			v[b].pb(i);//反向连接
		}
	}
	for(int i = 1;i <= n;i++)//重复子结数次搜索
	{
		sum = 0;//初始化路程和为0
		for(int j = 1;j <= n;j++)	vis[j] = 0;//在每次搜索前把所有的标记变成0
		dfs(i,1);//让第i个结点为根节点开始搜索（第二个参数1，表示深度为1，即根节点）
		minn = min(minn,sum);//用minn变量存入最小值
	}
	cout<<minn;//输出最小路程和
}