题目描述
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
输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。
输入输出样例
输入样例#1：
6 1 2 3 4 5
输出样例#1：
2
代码
1 #include<iostream> 2 #include<cstring> 3 #include<cstdio> 4 #include<algorithm> 5 #include<vector> 6 using namespace std; 7 8 int N,p=1,ans,x,vis[5005],pre[5005]; 9 vector<int> G[5005]; 10 11 struct cc{ 12 int num,dis; 13 }nod[5005]; 14 bool cmp(cc a,cc b){ 15 return a.dis>b.dis; 16 } 17 void dfs(int x,int dep){ 18 vis[x]=1; 19 nod[p].dis=dep; 20 nod[p++].num=x; 21 for(int i=0;i<G[x].size();i++) 22 if(!vis[G[x][i]]){ 23 dfs(G[x][i],dep+1); 24 pre[G[x][i]]=x; 25 } 26 } 27 28 void paint(int x,int t){ 29 if(t<0||x<0) return; 30 vis[x]=1; 31 for(int i=0;i<G[x].size();i++) 32 // if(!vis[G[x][i]]){ 33 paint(G[x][i],t-1); 34 // } 35 } 36 37 int main(){ 38 // freopen("01.in","r",stdin); 39 scanf("%d",&N); 40 for(int i=2;i<=N;i++){ 41 scanf("%d",&x); 42 G[i].push_back(x); 43 G[x].push_back(i); 44 } 45 46 memset(pre,-1,sizeof(pre)); 47 48 dfs(1,0); 49 50 memset(vis,0,sizeof(vis)); 51 sort(nod+1,nod+N+1,cmp); 52 53 for(int i=1;i<=N;i++){ 54 x=nod[i].num; 55 if(!vis[x]){ 56 ++ans; 57 if(pre[x]<0) paint(x,0); 58 else if(pre[pre[x]]<0) paint(pre[x],1); 59 else paint(pre[pre[x]],2); 60 } 61 } 62 printf("%d ",ans); 63 return 0; 64 }Line 32~33是大坑，大雾。。
先把任意一个点设为根节点无根树转有根树，再把所有点按深度排序，按深度从小到大遍历每一个点，如果这个点还没有被覆盖，就把它的二级祖先设为一个消防局，并从这个消防局开始覆盖与它距离不超过2的结点（如果这个点没有二级祖先就把它的父节点设为消防局，如果这个点是根结点就直接把它设为消防局）。可以用调整法证明这个贪心的正确性，把它的父节点设为消防局覆盖的点一定没有把二级祖先设为消防局覆盖的节点多（想一想，为什么）。