消防局的设立
题解
看到这道题,相信许多人都会考虑树形dp。可惜笔者dp半天写不对,所以写的贪心。
贪心可以这样想:
我们先从深度最大的点想起,如果此时这个点无法覆盖,我们选择它的父亲的父亲,因为这个点是可以覆盖到这个点的可以覆盖最多点的点。这样从最深的点枚举起,枚举所有点。
源码
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<stack>
#include<vector>
using namespace std;
typedef long long LL;
const int INF=0x3f3f3f3f;
int n,sum[1005],ans;
int tot,head[1005];
struct point
{
int f[20];
int dep;
bool reach;
}p[1005];
struct ming
{
int from,to;
int next;
}edge[2005];
/*#define gc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<22,stdin),p1 == p2)?EOF:*p1++)
char buf[(1<<22)],*p1=buf,*p2=buf;*/
#define gc() getchar()
template<typename _T>
inline void read(_T &x)
{
int f=1;x=0;char s=gc();
while(s>'9'||s<'0'){if(s=='-')f=-1;s=gc();}
while(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=gc();}
x*=f;
}
void addEdge(int u,int v)
{
edge[++tot].from=u;edge[tot].to=v;
edge[tot].next=head[u];head[u]=tot;
}
bool check()
{
for(int i=1;i<=n;i++)
if(!p[i].reach)
return false;
return true;
}
void dfs(int u,int fa)
{
sum[u]=1;
p[u].dep=p[fa].dep+1;
p[u].f[0]=fa;
for(int i=1;i<18;i++)
p[u].f[i]=p[p[u].f[i-1]].f[i-1];
for(int i=head[u];i;i=edge[i].next)
{
int v=edge[i].to;
if(v==fa) continue;
dfs(v,u);sum[u]+=sum[v];
}
}
void built(int u)
{
for(int i=head[u];i;i=edge[i].next)
{
int v=edge[i].to;
p[v].reach=true;
for(int j=head[v];j;j=edge[j].next)
{
int w=edge[j].to;
p[w].reach=true;
}
}
}//建立消防站
int main()
{
read(n);
for(int i=2;i<=n;i++)
{
int v;read(v);
addEdge(i,v);
addEdge(v,i);
}
dfs(1,1);//搜索这棵树
while(!check())
{
int maxx=0,k=0;
for(int i=1;i<=n;i++)
if(p[i].dep>maxx&&!p[i].reach)
maxx=p[i].dep,k=i;//寻找当前最深的未覆盖的点
built(p[k].f[1]);ans++;//选择它父亲的父亲
}
printf("%d",ans);
return 0;
}