[HNOI2003]消防局的设立

OJ题号：BZOJ1217、洛谷2279

思路：贪心。

先O(n)递推记录各个结点的深度，然后从深度大的结点贪心，如果当前结点不安全，就在它爷爷地方开消防局，同时更新上下二代的安全信息。

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<functional>
const int N=1001;
int p[N];
bool safe[N]= {0};
std::vector<int> c[N];
struct Vertex {
    int id,depth;
    bool operator > (const Vertex &x) const {
        return this->depth>x.depth;
    }
};
Vertex v[N];
void update(const int x) {
    safe[p[x]]=safe[p[p[x]]]=true;
    for(unsigned int i=0; i<c[p[x]].size(); i++) safe[c[p[x]][i]]=true;
    for(unsigned int i=0; i<c[x].size(); i++) {
        safe[c[x][i]]=true;
        for(unsigned int j=0; j<c[c[x][i]].size(); j++) {
            safe[c[c[x][i]][j]]=true;
        }
    }
}
int main() {
    int n;
    scanf("%d",&n);
    p[1]=1;
    v[1].id=1;
    v[1].depth=0;
    for(int i=2; i<=n; i++) {
        v[i].id=i;
        scanf("%d",&p[i]);
        c[p[i]].push_back(i);
        v[i].depth=v[p[i]].depth+1;
    }
    std::sort(&v[1],&v[n+1],std::greater<Vertex>());
    int ans=0;
    for(int i=1; i<=n; i++) {
        if(safe[v[i].id]) continue;
        update(p[p[v[i].id]]);
        ans++;
    }
    printf("%d
",ans);
    return 0;
}