[HNOI2003]消防局的设立

== \(\color{yellow}{只写给自己看防止自己忘了的emmm}\) ==

主要思路是从叶节点开始向根节点进行转移

一共有五个状态,f[i][0...4]

0 状态表示当前节点为消防站,可以管理以i节点的子树以及i的爷爷
1 状态表示当前节点的儿子v为消防站,可以管理以v为根的子树和i号节点的爸爸
2 状态表示当前节点的孙子t为消防站,可以管理以t为根的子树和i号节点的爷爷
3 状态表示当前节点的重孙子ct为消防站,可以管理以ct为根的子树和ct节点的儿子
4 ...........

这是一个好博客(题解)!

code

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define maxn 2100
using namespace std ;
int read() {
    int x = 0 , f = 1 ; char s = getchar() ;
    while(s > '9' || s < '0') {if(s == '-') f = -1 ; s = getchar() ;}
    while(s <='9' && s >='0') {x = x * 10 + (s-'0'); s = getchar() ;}
    return x*f ;
}
struct dy{
    int x , y , z , next ;
}a[maxn] ;
int f[maxn][10] , n ;
int head[maxn] , vis[maxn] , dep[maxn] , t ; 
int add(int x ,int y) {
    a[++t].x = x ;
    a[t].y = y ;
    a[t].z = 1 ;
    a[t].next = head[x] ;
    head[x] = t ;   
}
void dfs(int now) {
    f[now][0] = 1 ;
    f[now][3] = 0 ;
    f[now][4] = 0 ;
    for(int i = head[now] ; i ; i = a[i].next) {
        int v = a[i].y ;
        dfs(v) ;
        f[now][0] += f[v][4] ;
        f[now][3] += f[v][2] ;
        f[now][4] += f[v][3] ;
    }
    if(!head[now]) {
        f[now][1] = f[now][2] = 1 ;
    }else {
        f[now][1] = f[now][2] = maxn * 10 ;
        for(int i = head[now] ; i ; i = a[i].next) {
            int v = a[i].y ;
            for(int j = head[now] ; j ; j = a[j].next) {
                if(i==j) continue ;
                int t = a[j].y ;
                f[v][0] += f[t][3] ;
                f[v][1] += f[t][2] ;
            }
            f[now][1] = min(f[now][1],f[v][0]) ;
            f[now][2] = min(f[now][2],f[v][1]) ;
        }
    }
    for(int i = 1 ; i <= 4 ; i ++) {
        f[now][i] = min(f[now][i],f[now][i-1]) ;
    }
}  
int main () {
    n = read() ;
    for(int i = 2 ; i <= n ; i ++) {
        int x = read() ;
    //  add(i,x) ;
        add(x,i) ;
    }
    dfs(1) ;
    printf("%d\n",f[1][2]) ;
    return 0 ;
} 

转载于:https://www.cnblogs.com/lyt020321/p/11354356.html