本文共 2079 字,大约阅读时间需要 6 分钟。
链接:https://ac.nowcoder.com/acm/problem/20031
来源:牛客网题目描述
2020年,人类在火星上建立了一个庞大的基地群,总共有n个基地。起初为了节约材料,人类只修建了n-1条道路来连接这些基地,并且每两个基地都能够通过道路到达,所以所有的基地形成了一个巨大的树状结构。如果基地A到基地B至少要经过d条道路的话,我们称基地A到基地B的距离为d。 由于火星上非常干燥,经常引发火灾,人类决定在火星上修建若干个消防局。消防局只能修建在基地里,每个消防局有能力扑灭与它距离不超过2的基地的火灾。你的任务是计算至少要修建多少个消防局才能够确保火星上所有的基地在发生火灾时,消防队有能力及时扑灭火灾。 输入描述: 输入文件的第一行为n,表示火星上基地的数目。 接下来的n-1行每行有一个正整数,其中文件第i行的正整数为a[i],表示从编号为i的基地到编号为a[i]的基地之间有一条道路,为了更加简洁的描述树状结构的基地群,有a[i] 输出描述: 输出文件仅有一个正整数,表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。 示例1 输入 复制 6 1 2 3 4 5 输出 复制 2 这是在树形dp环节见到的这个题目,但是树形dp偷看了一下题解,有点小麻烦。。就换成贪心来做了。我们对于一条链上的节点来说,在叶子节点建立基地的效果一定没有在他的祖父节点建立基地的效果好。因为他的祖父节点还可以往上惠泽上面的节点。这就是贪心的思想。我们dfs一遍出来的是各个节点的深度和他们各自的父亲节点。按着深度排序,用刚才的想法去贪心求就可以了。 代码如下:#include#define ll long longusing namespace std;const int maxx=1e5+100;struct edge{ int to,next;}e[maxx<<1];int head[maxx<<1],fa[maxx],deep[maxx],a[maxx],dis[maxx];int tot,n;/*-----------事前准备-------------*/inline void init(){ memset(dis,-1,sizeof(dis)); memset(head,-1,sizeof(head)); tot=0;}inline void add(int u,int v){ e[tot].to=v,e[tot].next=head[u],head[u]=tot++;}inline bool cmp(int a,int b){ return deep[a]>deep[b];//按着深度由大到小排序}/*-------------dfs------------*/inline void dfs(int u,int f){ deep[u]=deep[f]+1; fa[u]=f; for(int i=head[u];i!=-1;i=e[i].next) { int to=e[i].to; if(to==f) continue; dfs(to,u); }}/*--------------贪心--------------*/inline int solve(){ int s,f,ff,fff,ffff,ans=0; for(int i=1;i<=n;i++) a[i]=i; sort(a+1,a+1+n,cmp); fa[0]=n+1,fa[fa[0]]=n+2;//设置虚父节点,要求出0的父亲节点还有祖父节点,否则会出错。 for(int i=1;i<=n;i++) { s=a[i];f=fa[s];ff=fa[f];fff=fa[ff];ffff=fa[fff];//求出这个点的父亲,祖父,曾祖父,曾曾祖父。 if(dis[s]>=0||dis[f]>=1||dis[ff]>=2) continue;//这是在寻找能够覆盖到它的条件 ans++; dis[s]=max(dis[s],0); dis[f]=max(dis[f],1); dis[ff]=max(dis[ff],2); dis[fff]=max(dis[fff],1); dis[ffff]=max(dis[ffff],0);//如果覆盖不到,就在它的祖父节点放置一个基地。 } return ans;}int main(){ int x; while(~scanf("%d",&n)) { init(); for(int i=2;i<=n;i++) { scanf("%d",&x); add(x,i),add(i,x); } deep[0]=0; dfs(1,0); printf("%d\n",solve()); } return 0;}
努力加油a啊,(o)/~
转载地址:http://kutvi.baihongyu.com/