【贪心】消防局的设立

第三题：消防局的设立

(fire.exe)

 

2020年，人类在火星上建立了一个庞大的基地群，总共有n个基地。起初为了节约材料，人类只修建了n-1条道路来连接这些基地，并且每两个基地都能够通过道路到达，所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话，我们称基地A到基地B的距离为d。

       由于火星上非常干燥，经常引发火灾，人类决定在火星上修建若干个消防局。消防局只能修建在基地里，每个消防局有能力扑灭与它距离不超过2的基地的火灾。

       你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时，消防队有能力及时扑灭火灾。

 

输入格式：

l      输入文件名为fire.in。

l      输入文件的第一行为n（n<1000），表示火星上基地的数目。接下来的n-1行每行有一个正整数，其中文件第i行的正整数为a[i]，表示从编号为i的基地到编号为a[i]的基地之间有一条道路，为了更加简洁的描述树状结构的基地群，有a[i]<i。

 

输出格式：

l       输出文件名为fire.out

l       输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。

 

输入输出示例：

fire.in

6

1

2

3

4

5

fire.out

2

这是一道很简单的贪心题目。

可是自己却想复杂了，虽然很大一部分原因是S哥的误导。。

一开始被告诉是树形DP，搞了一晚上。。

然后被告诉是搜索，搞了一晚上。。

然后发现是贪心。。

可以感觉得到的是，离根较远的点更有机会覆盖到更多的点，同时，我们不取最远的点也不取最远的点的父亲，而取爷爷。因为取爷爷，是可以覆盖到叶节点的，同时可以覆盖到更多其他的点。

每次找到离根最远的叶节点，用堆来维护就行了。找到他的爷爷节点，然后以他爷爷节点开始删除距离小于等于2的点，操作的次数就是答案。

一开始我以为要删除，所以用双向链表，但是一直搞不好。。最后发现不用删除，只用打一个删除标记就行了。

#include <cstdio>
#include <queue>

struct Leaf
{
	long ind;
	long depth;
	Leaf(long i,long d):ind(i),depth(d){}
	bool operator<(const Leaf& l2)const{return depth<l2.depth;}
};
std::priority_queue<Leaf> leaf;

struct node
{
	long ind;
	node* nxt;
};
node* head[1010];

long depth[1010];
long fa[1010];
bool deleted[1010];
long n;

void dfs(long u,long d)
{
	depth[u] = d;
	for (node* vv=head[u];vv;vv=vv->nxt)
		if (vv->ind-fa[u])
			dfs(vv->ind,d+1);
}
void del(long u,long d,long last)
{
	deleted[u] = true;
	if (d == 2)
	{
		return;
	}
	for (node* vv=head[u];vv;vv=vv->nxt)
		if (last != vv->ind)
			del(vv->ind,d+1,u);
}
long getint()
{
	long rs=0;char tmp;bool sgn=1;
	do tmp = getchar();
	while (!isdigit(tmp)&&tmp-'-');
	if (tmp=='-'){sgn=0;tmp=getchar();}
	do rs=(rs<<3)+(rs<<1)+tmp-'0';
	while (isdigit(tmp=getchar()));
	return sgn?rs:-rs;
}
void insert(long a,long b)
{
	node* nn = new node;
	nn->ind = b;
	nn->nxt = head[a];
	head[a] = nn;
}
int main()
{
	freopen("fire.in","r",stdin);
	freopen("fire.out","w",stdout);

	n = getint();
	for (long a=2;a<n+1;a++)
	{
		long b = getint();
		insert(b,a);
		insert(a,b);
		fa[a] = b;
	}
	dfs(1,1);
	for (long i=1;i<n+1;i++)
	{
		leaf.push(Leaf(i,depth[i]));
	}
	long ans = 0;
	while (!leaf.empty())
	{
		long u = -1;
		long tmp = 0;
		do {
			tmp = leaf.top().ind;
			leaf.pop();
			if (!deleted[tmp])
				u = tmp;
		}while (!leaf.empty()&&u==-1);
		if (u == -1)
			break;

		if (fa[fa[u]])
			u = fa[fa[u]];
		else if (fa[u])
			u = fa[u];
		del(u,0,0);
		ans ++;
	}
	printf("%ld",ans);
	return 0;
}