题目描述
2020年,人类在火星上建立了一个庞大的基地群,总共有n个基地。起初为了节约材料,人类只修建了n-1条道路来连接这些基地,并且每两个基地都能够通过道路到达,所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话,我们称基地A到基地B的距离为d。
由于火星上非常干燥,经常引发火灾,人类决定在火星上修建若干个消防局。消防局只能修建在基地里,每个消防局有能力扑灭与它距离不超过2的基地的火灾。
你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时,消防队有能力及时扑灭火灾。
输入输出格式
输入格式:输入文件名为input.txt。
输入文件的第一行为n (n<=1000),表示火星上基地的数目。接下来的n-1行每行有一个正整数,其中文件第i行的正整数为a[i],表示从编号为i的基地到编号为a[i]的基地之间有一条道路,为了更加简洁的描述树状结构的基地群,有a[i]<i。
输出格式:输出文件名为output.txt
输出文件仅有一个正整数,表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。
每个消防站可以扑灭与它距离不超过2的点的火灾,所以每个叶子结点能被消防站覆盖,有三种情况:这个点的兄弟结点,父亲结点和爷爷结点。显然,我们应该选择爷爷结点,因为父亲结点和兄弟结点覆盖的点,爷爷结点都能覆盖到,所以爷爷结点覆盖的点一定多于它们覆盖的点。
我们每次找出深度最大的结点,将覆盖到的点标记为已覆盖,这样做贪心即可。
暴力判断是否已经达到要求和找深度最大结点就可以,复杂度O(n^2)。
代码:
#include<cstdio> #include<iostream> #include<algorithm> #include<cstring> #include<cmath> using namespace std; struct Edge { int to,next; }e[1010]; int head[1010]; void insert(int from,int to) { static int c=0; e[++c]=(Edge){to,head[from]}; head[from]=c; } int fa[1010]; bool vis[1010]; int dep[1010]; void dfs(int now,int d) { vis[now]=false; if (d==0) return; for(int i=head[now]; i!=0; i=e[i].next) { dfs(e[i].to,d-1); } dfs(fa[now],d-1); } int main() { int n; scanf("%d",&n); fa[1]=1; for(int i=2; i<=n; ++i) { int a; scanf("%d",&a); insert(a,i); fa[i]=a; } for(int i=1; i<=n; ++i) { dep[i]=dep[fa[i]]+1; } int ans=0; memset(vis,true,sizeof(vis)); while(1) { int t=0; for(int i=1; i<=n; ++i) { if (vis[i]&&dep[i]>dep[t]) t=i; } if (t==0) break; dfs(fa[fa[t]],2); ans++; } cout<<ans; return 0; }本题可以用堆来实现取深度最大结点,优化为O(nlogn),代码大家可以自己实现。