【8.8gzoj综合】【洛谷P1041】传染病控制【DFS】

题意

给定一棵树，可以在每一层（除第一层）删去一颗以某个节点为根的树，问最后剩下的结点的最少个数。

分析

这道题$n<=300$，可以考虑直接暴力搜索。

预处理出每每个点的儿子和每层的点。然后枚举每一层和每个点，处理出子树中点的感染情况，用个数组标记，然后下一层。

其实就是搜索出了所有可能的感染情况，取min就行。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int n,p,tr[310][310],d[310][310];//d[i][j]表示第i层的第j个人是谁 
int f[310];

void dfs(int x,int k)//标记每一层和人 
{
	d[k][++d[k][0]]=x;
	if(!tr[x][0]) return;
	for(int i=1;i<=tr[x][0];i++)//遍历所有儿子 
	{
		dfs(tr[x][i],k+1);
	}
	return;
}

int clear(int x,int k)//可以恢复或者取消感染，取决于k值 
{
	if(k) f[x]=0;
	else f[x]=1;
	if(!tr[x][0]) return 1;
	int s=1;
	for(int i=1;i<=tr[x][0];i++)
	{
		s+=clear(tr[x][i],k);
	}
	return s;
}

int getans(int x,int ans)
{
	if(!d[x][0]) return ans;
	int mn=0x3f3f3f3f,tmp;
	int ff=0;
	for(int i=1;i<=d[x][0];i++)
	{
		if(f[d[x][i]]) continue;
		ff=1;
		tmp=clear(d[x][i],0);//子树人数都取消感染 
		mn=min(getans(x+1,ans-tmp),mn);//下一层 
		clear(d[x][i],1);//回溯 
	}
	if(!ff) return ans;//如果这一层都没有能被感染的人 
	return mn;
}

int main()
{
    cin>>n>>p;
	for(int i=1;i<=p;i++)
	{
		int x,y;
		cin>>x>>y;
		if(x>y) swap(x,y);
		tr[x][++tr[x][0]]=y;//tr[x][0]是x的儿子个数 
	}	
	dfs(1,1);
	cout<<getans(2,n);//假设一开始全部被感染 
	return 0;
}