消防局的设立

消防局的设立

题解

看到这道题，相信许多人都会考虑树形dp。可惜笔者dp半天写不对，所以写的贪心。

贪心可以这样想：

我们先从深度最大的点想起，如果此时这个点无法覆盖，我们选择它的父亲的父亲，因为这个点是可以覆盖到这个点的可以覆盖最多点的点。这样从最深的点枚举起，枚举所有点。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream> 
#include<algorithm>
#include<stack>
#include<vector>
using namespace std;
typedef long long LL;
const int INF=0x3f3f3f3f;
int n,sum[1005],ans;
int tot,head[1005];
struct point
{
	int f[20];
	int dep;
	bool reach;
}p[1005];
struct ming
{
	int from,to;
	int next;
}edge[2005];
/*#define gc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<22,stdin),p1 == p2)?EOF:*p1++)
char buf[(1<<22)],*p1=buf,*p2=buf;*/
#define gc() getchar()
template<typename _T>
inline void read(_T &x)
{
	int f=1;x=0;char s=gc();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=gc();}
	while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=gc();}
	x*=f;
}
void addEdge(int u,int v)
{
	edge[++tot].from=u;edge[tot].to=v;
	edge[tot].next=head[u];head[u]=tot;
}
bool check()
{
	for(int i=1;i<=n;i++)
		if(!p[i].reach)
			return false;
	return true;
}
void dfs(int u,int fa)
{
	sum[u]=1;
	p[u].dep=p[fa].dep+1;
	p[u].f[0]=fa;
	for(int i=1;i<18;i++)
		p[u].f[i]=p[p[u].f[i-1]].f[i-1];
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa) continue;
		dfs(v,u);sum[u]+=sum[v];
	}
}
void built(int u)
{
	for(int i=head[u];i;i=edge[i].next)
	{
		int v=edge[i].to;
		p[v].reach=true;
		for(int j=head[v];j;j=edge[j].next)
		{
			int w=edge[j].to;
			p[w].reach=true;
		}
	}
}//建立消防站
int main()
{
	read(n);
	for(int i=2;i<=n;i++)
	{
		int v;read(v);
		addEdge(i,v);
		addEdge(v,i);	
	}
	dfs(1,1);//搜索这棵树
	while(!check())
	{
		int maxx=0,k=0;
		for(int i=1;i<=n;i++)
			if(p[i].dep>maxx&&!p[i].reach)
				maxx=p[i].dep,k=i;//寻找当前最深的未覆盖的点
		built(p[k].f[1]);ans++;//选择它父亲的父亲
	}
	printf("%d",ans);
	return 0;
}

谢谢！！！