【BZOJ 1217】[HNOI2003]消防局的设立（贪心）

最新推荐文章于 2021-11-18 22:01:30 发布

原创最新推荐文章于 2021-11-18 22:01:30 发布 · 646 阅读

0 ·

CC 4.0 BY-SA版权

贪心专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种使用图论解决消防站最优选址问题的方法。通过定义f[i]为节点i到最近消防站的距离，并利用DFS遍历进行计算，确保每个节点到最近消防站的距离不超过规定值。

题目：我是超链接

题解：
f[i]为i到达最近消防站的距离，对于一个叶子节点，用了一些特殊操作使其在2格之内必定建立消防站

代码：

#include <cstdio>
#include <iostream>
#define INF 1e9
#define N 1005
using namespace std;
int tot,nxt[N*2+5],point[N*2+5],v[N*2+5],f[N*2+5],ans;
void addline(int x,int y)
{
	++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
	++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int now,int fa)
{
	int maxx=-INF,minn=INF;
	for (int i=point[now];i;i=nxt[i])
	  if (v[i]!=fa)
	  {
	  	dfs(v[i],now);
	  	minn=min(minn,f[v[i]]);
	  	maxx=max(maxx,f[v[i]]);
	  }
	if (minn+maxx<=3) f[now]=minn+1;
	else f[now]=maxx+1;
	if (maxx==-INF) f[now]=3;//叶子节点 
	if (f[now]==5){ans++; f[now]=0;}
    if (f[now]>=3 && fa==0) ans++;
}
int main()
{
	int n,i;
	scanf("%d",&n);
	for (i=1;i<n;i++)
	{
		int a;
		scanf("%d",&a);
		addline(i+1,a);
	}
	dfs(1,0);
	printf("%d",ans);
}