https://www.luogu.org/problemnew/show/P2279
一开始就想到了贪心的方法，不过一直觉得不能证明。
贪心的考虑是在深度从深到浅遍历每个结点的过程中，对于每个没有覆盖的结点选择覆盖他的祖父结点。
仔细想想觉得这是正确的。
在实现的过程中有一个小技巧是o[i]记录i结点距离消防局最近的距离，如果o[i] > 2则需要在他的祖父结点建立一个消防站。用这种方法可以很方便的判断兄弟节点是否被覆盖。
一个细节是要给根节点1建立两个虚结点N + 1和N + 2作为他的父亲结点和祖父结点，不然在1结点寻找祖父节点的时候容易出现问题。
用这样的方法也可以在常数很小的时间内实现树上半径k的覆盖问题
#include <map>
#include <set>
#include <ctime>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
inline int read(){int now=;register char c=getchar();for(;!isdigit(c);c=getchar());
for(;isdigit(c);now=now*+c-'',c=getchar());return now;}
#define For(i, x, y) for(int i=x;i<=y;i++)
#define _For(i, x, y) for(int i=x;i>=y;i--)
#define Mem(f, x) memset(f,x,sizeof(f))
#define Sca(x) scanf("%d", &x)
#define Sca2(x,y) scanf("%d%d",&x,&y)
#define Sca3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define Scl(x) scanf("%lld",&x);
#define Pri(x) printf("%d\n", x)
#define Prl(x) printf("%lld\n",x);
#define CLR(u) for(int i=0;i<=N;i++)u[i].clear();
#define LL long long
#define ULL unsigned long long
#define mp make_pair
#define PII pair<int,int>
#define PIL pair<int,long long>
#define PLL pair<long long,long long>
#define pb push_back
#define fi first
#define se second
typedef vector<int> VI;
const double eps = 1e-;
const int maxn =;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + ;
int N,M,K;
int fa[maxn];
int deep[maxn];
int a[maxn];
int o[maxn];
bool cmp(int a,int b){
return deep[a] > deep[b];
}
int main(){
Sca(N); o[] = INF;
fa[] = N + ;
fa[N + ] = N + ;
o[N + ] = o[N + ] = INF;
for(int i = ; i <= N; i ++) a[i] = i;
for(int i = ; i <= N ; i ++){
Sca(fa[i]);
deep[i] = deep[fa[i]] + ;
o[i] = INF;
}
sort(a + ,a + + N,cmp);
int ans = ;
for(int i = ; i <= N ; i ++){
int u = a[i];
int v = fa[u],w = fa[fa[u]];
o[u] = min(o[u],min(o[v] + ,o[w] + ));
if(o[u] > ){
o[w] = ;
o[fa[w]] = min(o[fa[u]],);
o[fa[fa[w]]] = min(o[fa[fa[w]]],);
ans++;
}
}
Pri(ans);
return ;
}
贪心
当然这道题出现在dp专题里，事实上也是一个很硬核的树dp，在没有想到或者证明贪心的时候，这样明显的树dp也不失为一种做法。
思路借鉴了luogu的题解，开始对于这样的树dp一直想着两次树dp，第一次记录根节点的answer，第二次利用根节点拓展到整棵树的answer，导致开始的思路仅限于dp[maxn][3]来记录这个点，这个点的儿子，这个点的孙子的消防站情况，使得第一次dfs就遇到了问题，可能可以做，但有些麻烦。
看了题解之后茅塞顿开，事实上树dp并不需要套路的总是两边dp，换一种思路同时维护两端的情况，就是除了原本的3个状态以外，重新记录dp[i][3]表示所有儿子都被覆盖的情况以及dp[i][4]所有孙子都被覆盖的情况。
仅用一遍dfs就可以解决这个问题。
#include <map>
#include <set>
#include <ctime>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include<algorithm>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <functional>
using namespace std;
inline int read(){int now=;register char c=getchar();for(;!isdigit(c);c=getchar());
for(;isdigit(c);now=now*+c-'',c=getchar());return now;}
#define For(i, x, y) for(int i=x;i<=y;i++)
#define _For(i, x, y) for(int i=x;i>=y;i--)
#define Mem(f, x) memset(f,x,sizeof(f))
#define Sca(x) scanf("%d", &x)
#define Sca2(x,y) scanf("%d%d",&x,&y)
#define Sca3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define Scl(x) scanf("%lld",&x);
#define Pri(x) printf("%d\n", x)
#define Prl(x) printf("%lld\n",x);
#define CLR(u) for(int i=0;i<=N;i++)u[i].clear();
#define LL long long
#define ULL unsigned long long
#define mp make_pair
#define PII pair<int,int>
#define PIL pair<int,long long>
#define PLL pair<long long,long long>
#define pb push_back
#define fi first
#define se second
typedef vector<int> VI;
const double eps = 1e-;
const int maxn = ;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + ;
int N,M,K;
struct Edge{
int to,next;
}edge[maxn * ];
int head[maxn],tot;
void init(){
for(int i = ; i <= N ; i ++) head[i] = -;
tot = ;
}
void add(int u,int v){
edge[tot].to = v;
edge[tot].next = head[u];
head[u] = tot++;
}
int dp[maxn][]; // 0这是,1儿子是,2孙子是,3儿子全被覆盖,4孙子全被覆盖
int Min(int a,int b){
return a > b?b:a;
}
int Min(int a,int b,int c){
return Min(Min(a,b),c);
}
int Min(int a,int b,int c,int d){
return Min(Min(a,b,c),d);
}
int Min(int a,int b,int c,int d,int e){
return Min(Min(a,b,c),Min(d,e));
}
void dfs(int t){
if(head[t] == -){
dp[t][] = ;
dp[t][] = dp[t][] = INF;
dp[t][] = dp[t][] = ;
return;
}
dp[t][] = ;
int MAXSON = INF;
int MAXGS = INF;
for(int i = head[t]; ~i; i = edge[i].next){
int v = edge[i].to;
dfs(v);
dp[t][] += Min(dp[v][],dp[v][],dp[v][],dp[v][],dp[v][]);
dp[t][] += Min(dp[v][],dp[v][],dp[v][],dp[v][]);
MAXSON = Min(dp[v][] - Min(dp[v][],dp[v][],dp[v][],dp[v][]),MAXSON);
MAXGS = Min(dp[v][] - Min(dp[v][],dp[v][],dp[v][]),MAXGS);
dp[t][] += Min(dp[v][],dp[v][],dp[v][]);
dp[t][] += Min(dp[v][],dp[v][],dp[v][]);
dp[t][] += Min(dp[v][],dp[v][],dp[v][],dp[v][]);
}
dp[t][] += MAXSON;
dp[t][] += MAXGS;
}
int main(){
Sca(N); init();
for(int i = ; i <= N; i ++){
int x; Sca(x);
add(x,i);
}
dfs();
cout << Min(dp[][],dp[][],dp[][]);
return ;
}