P2279 [HNOI2003]消防局的设立

P2279 [HNOI2003]消防局的设立
考场上想出了贪心策略，但是处理细节时有点问题，gg了。从（当前深度最大的节点）叶子节点往上跳k个，在这里设消防局，并从消防局遍历k个距离，标记上。

 #include<iostream>

 #include<cstdio>

 #include<queue>

 #include<algorithm>

 #include<cmath>

 #include<ctime>

 #include<set>

 #include<map>

 #include<stack>

 #include<cstring>

 #define inf 2147483647

 #define For(i,a,b) for(register int i=a;i<=b;i++)

 #define p(a) putchar(a)

 #define g() getchar()

 //by war

 //2017.11.1

 using namespace std;

 int n,x,k=;

 int d[];

 int cnt;

 int ans;

 bool vis[];

 struct node

 {

     int n;

     node *next;

 }*e[];

 struct nod

 {

     int deep;

     bool operator<(const nod&aa)const

     {

         return deep>aa.deep;

     }

     int pos;

 }a[];

 void in(int &x)

 {

     int y=;

     char c=g();x=;

     while(c<''||c>'')

     {

     if(c=='-')

     y=-;

     c=g();

     }

     while(c<=''&&c>='')x=(x<<)+(x<<)+c-'',c=g();

     x*=y;

 }

 void o(int x)

 {

     if(x<)

     {

         p('-');

         x=-x;

     }

     if(x>)o(x/);

     p(x%+'');

 }

 void push(int x,int y)

 {

     node *p;

     p=new node();

     p->n=y;

     if(e[x]==NULL)

     e[x]=p;

     else

     {

         p->next=e[x]->next;

         e[x]->next=p;

     }

 }

 void dfs(int x,int dis)

 {

     if(dis>k)

     return;

     vis[x]=true;

     cnt--;

     for(node *i=e[x];i!=NULL;i=i->next)

     dfs(i->n,dis+);

 }

 int main()

 {

     in(n);

     cnt=n;

     For(i,,n)

     {

         in(x);

         push(x,i);

         push(i,x);

         a[i].deep=a[x].deep+;

         d[i]=x;

         a[i].pos=i;

     }

     a[].pos=;

     d[]=;

     sort(a+,a+n+);

     while(cnt>)

     {

         For(i,,n)

         if(!vis[a[i].pos])

         {

             x=a[i].pos;

             For(j,,k)

             x=d[x];

             dfs(x,);

             ans++;

         }

     }

     o(ans);

      return ;

 }