题目描述
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
貌似不像是dp,更多的是贪心……
因为是树,所以我们每次考虑最深点,考虑在他爷爷节点安放(注意可能没有爷爷),然后再扫一遍就过了。
好久没用STL,今天试了试,很棒。
AC代码如下:
#include<cstdio> #include<algorithm> #include<vector> #include<queue> using namespace std; const int N=1000+5; int fa[N],n,x,dep[N],ans; bool vis[N]; vector<int>s[N]; struct cmp { bool operator () (const int &a,const int &b) { return dep[a]<dep[b]; } }; priority_queue<int,vector<int>,cmp>q; void dfs(int u,int from,int step) { fa[u]=from; dep[u]=step; for(int i=0;i<s[u].size();i++) if(s[u][i]!=from) dfs(s[u][i],u,step+1); } void solve(int u,int step) { if(step>2) return; vis[u]=1; for(int i=0;i<s[u].size();i++) solve(s[u][i],step+1); } int main() { scanf("%d",&n); for(int i=2;i<=n;i++) scanf("%d",&x),s[i].push_back(x),s[x].push_back(i); dfs(1,0,1); for(int i=1;i<=n;i++) q.push(i); while(q.size()) { while(q.size()&&vis[q.top()]) q.pop(); if(q.empty()) break; x=q.top(); q.pop(); if(fa[fa[x]]) solve(fa[fa[x]],0); else solve(x,0); ans++; } printf("%d",ans); return 0; }