洛谷2279消防局的设立

题目描述

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]<i。

输出格式：
输出文件名为output.txt

输出文件仅有一个正整数，表示至少要设立多少个消防局才有能力及时扑灭任何基地发生的火灾。

输入输出样例

输入样例#1：
6
1
2
3
4
5
输出样例#1：
2

---

稍复杂的树形DP
对于节点u
f[u][0]表示u设为消防局
f[u][1]表示在以u为根的子树中有到u距离<=1的点
f[u][2]表示在以u为根的子树中有到u距离<=2的点
…
f[u][4]表示在以u为根的子树中有到u距离<=4的点
易得:
f[i][0] = 1+Σmin(f[j][0…4])
f[i][1] = min( f[k][0] + Σ(j != k)min(f[j][0…3]) )
f[i][2] = min( f[k][1] + Σ(j != k)min(f[j][0…2]) )
f[i][3] = Σf[j][0…2];f[i][4] = Σf[j][0…3];
简化得
f[i][0] = 1+Σf[j][4]
f[i][1] =Σ(k)(f[k][0] + Σ(j != k)f[j][3]) =f[i][4] + min(Σ(k)(f[k][0] - f[k][3]))
f[i][2] =f[i][3] + min(f[k][1] - f[k][2])
f[i][3],f[i][4]略(自己看代码)
最后记得要
for(int i = 1; i <= 4; i ++)
f[u][i] = min(f[u][i], f[u][i - 1]);

贴代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxN = 1000;
int head[maxN + 1];
int cnt;
struct edge
{
    int v, next;
}G[maxN << 1];
long long f[maxN + 1][5];
void addEdge(int u, int v)
{
    G[cnt].v = v;
    G[cnt].next = head[u];
    head[u] = cnt ++;
}
void DFS(int u, int fa)
{
    int flag = 0;
    f[u][1] = (int)1e9;
    f[u][2] = (int)1e9;
    for(int i = head[u]; i != - 1; i = G[i].next)
        if(G[i].v != fa)
        {
            flag = 1;
            int v = G[i].v;
            DFS(v, u);
            f[u][0] += f[v][4];
            f[u][1] = min(f[u][1], f[v][0] - f[v][3]);
            f[u][2] = min(f[u][2], f[v][1] - f[v][2]);
            f[u][3] += f[v][2];
            f[u][4] += f[v][3];
        }
    if(! flag)
    {
        f[u][0] = f[u][1] = f[u][2] = 1;
        f[u][3] = f[u][4] = 0;
        return;
    }
    f[u][0] ++;
    f[u][1] += f[u][4];
    f[u][2] += f[u][3];
    for(int i = 1; i <= 4; i ++)
        f[u][i] = min(f[u][i], f[u][i - 1]);
}
int main()
{
    int n;
    scanf("%d", &n);
    memset(head, -1, sizeof(head));
    cnt = 0;
    for(int i = 2; i <= n; i ++)
    {
        int v;
        scanf("%d", &v);
        addEdge(i, v);
        addEdge(v, i);
    }
    memset(f, 0, sizeof(f));
    DFS(1, - 1);
    printf("%d", f[1][2]);
}