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

这道题目的背景是传染病控制,其传播路径呈树状结构,每轮只感染一代人。给定一棵树形结构表示传播途径,要求找出一个顺序来切断一条边,以使感染人数最少。输入包含树的节点数n和边数p,以及具体的边信息。程序的目标是输出在采取措施后的总感染人数。

题目链接
传染病的传播具有两种很特殊的性质;
第一是它的传播途径是树型的,一个人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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值