洛谷P1041 传染病控制(建树+dfs)

题目链接
传染病的传播具有两种很特殊的性质;
第一是它的传播途径是树型的,一个人X只可能被某个特定的人Y感染,只要Y不得病,或者是XY之间的传播途径被切断,则X就不会得病。
第二是,这种疾病的传播有周期性,在一个疾病传播周期之内,传染病将只会感染一代患者,而不会再传播给下一代。
这些性质大大减轻了蓬莱国疾病防控的压力,并且他们已经得到了国内部分易感人群的潜在传播途径图(一棵树)。但是,麻烦还没有结束。由于蓬莱国疾控中心人手不够,同时也缺乏强大的技术,以致他们在一个疾病传播周期内,只能设法切断一条传播途径,而没有被控制的传播途径就会引起更多的易感人群被感染(也就是与当前已经被感染的人有传播途径相连,且连接途径没有被切断的人群)。当不可能有健康人被感染时,疾病就中止传播。所以,蓬莱国疾控中心要制定出一个切断传播途径的顺序,以使尽量少的人被感染。
你的程序要针对给定的树,找出合适的切断顺序。

输入格式:
第一行是两个整数n(1≤n≤300)和p。
接下来pp行,每一行有22个整数i和j,表示节点i和j间有边相连。(意即,第i人和第j人之间有传播途径相连)。其中节点1是已经被感染的患者。

输出格式:
1行,总共被感染的人数。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
using namespace std;
const int maxn=305;
int n,m,x,y,deep[maxn];
bool vis[maxn];
int tot;
int ans=0x3f3f3f;
vector<int>v[maxn],son[maxn];
void build(int i)//建树 
{
	vis[i]=1;
	for(int j=0;j<v[i].size();j++)
	{
		if(!vis[v[i][j]])
		{
			son[i].push_back(v[i][j]);
			build(v[i][j]);
		}
	}
}
void dfs(int level)
{
	bool flag=0;
	if(tot>=ans) return;
	for(int i=1;i<=n;i++)
	{
		if(deep[i]==level)//节点i在level层
		{
			for(int j=0;j<son[i].size();j++)
			{
				flag=1;//代表第level+1层有节点,即为能到达下一层
				tot++;
				deep[son[i][j]]=level+1;//节点i的孩子的deep值赋为le+1
			}
		}
	}//tot为此时感染人数 
	if(!flag)
	{
		ans=min(ans,tot); return;
	}
	else
	{
		tot--;
		for(int i=1;i<=n;i++)
		{
			if(deep[i]==level+1)
			{
				deep[i]=0;
				dfs(level+1);
				deep[i]=level+1;//回溯 
			}
		}
		tot++;//回溯 
		for(int i=1;i<=n;i++)
		{
			if(deep[i]==level+1)
			{
				deep[i]=0;
				tot--;
			}
		}
	}
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y;
		v[x].push_back(y);
		v[y].push_back(x);
	}
	tot=1;
	deep[1]=1;
	build(1);
	dfs(1);
	cout<<ans<<endl;
}
    原文作者:传染病问题
    原文地址: https://blog.csdn.net/Nan_Liu/article/details/83187912
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞