洛谷 P2279 [HNOI2003]消防局的设立

来源:互联网 发布:怎么破解软件 编辑:程序博客网 时间:2024/05/20 08:23

题目描述
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]

#include<iostream>#include<cstring>#include<cstdio>#include<algorithm>using namespace std;#define MAXN 5005struct Node{ int id,dep; }t[MAXN];struct Edge{ int to,next; }e[MAXN<<1];int n,Ans=0,head[MAXN],fa[MAXN],tot,dep[MAXN];bool vis[MAXN];inline void Add_Edge(int u,int v){ e[++tot].to=v,e[tot].next=head[u],head[u]=tot; }inline void read(int &x){    x=0; int f=1; char c=getchar();    while(c>'9'||c<'0'){ if(c=='-')f=-1; c=getchar(); }    while(c>='0'&&c<='9'){ x=x*10+c-'0'; c=getchar(); } x*=f;}void DFS(int u,int from){    for(int i=head[u];i;i=e[i].next){        int v=e[i].to;        if(v==from) continue;        dep[v]=dep[u]+1,fa[v]=u;        DFS(v,u);    }}inline bool cmp(Node a,Node b){ return a.dep > b.dep; }void Push_Down(int u,int f,int k){    if(!k) return;    for(int i=head[u];i;i=e[i].next){        int v=e[i].to;        if(v==f) continue;        vis[v]=true;        Push_Down(v,u,k-1);    }}int Main(){    read(n);    for(int x,i=2;i<=n;++i)        read(x),Add_Edge(x,i),Add_Edge(i,x);    dep[1]=0;    DFS(1,0);    for(int i=1;i<=n;++i) t[i].id=i,t[i].dep=dep[i];    std::sort(t+1,t+n+1,cmp);    fa[1]=1;    for(int i=1;i<=n;++i){        int u=t[i].id;        if(!vis[u]){            int p=fa[fa[u]];            vis[p]=true;            Push_Down(p,0,2);            ++Ans;        }    }    printf("%d\n",Ans);    return 0;}int Aptal_is_My_Son=Main();int main(int argc,char *argv[]){ ; }

贪心的思路:
对于某个叶节点,能覆盖到它的只有自己、父节点或者爷爷节点,肯定是在爷爷节点会更优,爷爷节点建立消防局之后对周围距离小于等于2的点染一下色。
按照深度从大到小排序(先处理叶节点)
动态规划虽然能做,但是还不如贪心好想、好些、跑得快

原创粉丝点击