1019: 算法提高 传染病控制
时间限制: 1 Sec 内存限制: 512 MB
提交: 8 解决: 4
题目描述
近来,一种新的传染病肆虐全球。蓬莱国也发现 了零星感染者,为防止该病在蓬莱国大范围流行,该国政府决定不惜一切代价控制传染病的蔓延。不幸的是,由于人们尚未完全认识这种传染病,难以准确判别病毒 携带者,更没有研制出疫苗以保护易感人群。于是,蓬莱国的疾病控制中心决定采取切断传播途径的方法控制疾病传播。经过 WHO(世界卫生组织)以及全球各国科研部门的努力,这种新兴传染病的传播途径和控制方法已经研究消楚,剩下的任务就是由你协助蓬莱国疾控中心制定一个有 效的控制办法。
研究表明,这种传染病的传播具有两种很特殊的性质;
第一是它的传播途径是树型的,一个人X只可能被某个特定的人Y感染,只要Y不得病,或者是XY之间的传播途径被切断,则X就不会得病。
第二是,这种疾病的传播有周期性,在一个疾病传播周期之内,传染病将只会感染一代患者,而不会再传播给下一代。
这些性质大大减轻了蓬莱国疾病防控的压力,并且他们已经得到了国内部分易感人群的潜在传播途径图(一棵树)。但是,麻烦还没有结束。由于蓬莱国疾控中 心人手不够,同时也缺乏强大的技术,以致他们在一个疾病传播周期内,只能设法切断一条传播途径,而没有被控制的传播途径就会引起更多的易感人群被感染(也 就是与当前已经被感染的人有传播途径相连,且连接途径没有被切断的人群)。当不可能有健康人被感染时,疾病就中止传播。所以,蓬莱国疾控中心要制定出一个 切断传播途径的顺序,以使尽量少的人被感染。你的程序要针对给定的树,找出合适的切断顺序。
输入
输入格式的第一行是两个整数n(1≤n≤300)和p。接下来p行,每一行有两个整数i和j,表示节点i和j间有边相连(意即,第i人和第j人之间有传播途径相连,注意:可能是i到j也可能是j到i)。其中节点1是已经被感染的患者。
对于给定的输入数据,如果不切断任何传播途径,则所有人都会感染。
输出
只有一行,输出总共被感染的人数。
样例输入
7 61 21 32 42 53 67 3
样例输出
3
解题思路:
首先代码不是自己写的,直接参考了别人,看了好多人的,结果代码比较难懂,而且没有注释,只有这一个人的博客看明白了,其链接如下
原文链接:参考代码来源博客链接
先上代码:
代码变量解释:
(1)变量m,n 即为题目中所说的n和p,分别是人数和关系数目
(2)v[310],t[310]为vector用的是容器,简单来说就是二维数组,做图论的题目时会可以用容器来存放图,比较简单明了,v使用来存放题目中所给数据的容器,为了把题目中
给的数据建成一个有向树,我们将使用v建成有向树t,然后在递归过程中对树t进行处理。将元素放入容器,用函数push_back(),求某容器中元素的个数,用size()函数。
(3)level数组用来存放节点node所在的层数。即level[node]中的值为节点node在树种的层数。
(4)vis数组是为无向树转化为有向树所开辟的数据,如果节点node处理过了,则vis[node]标记为1,初始时整个数组赋值为0,代表所有节点都还未处理。
(5)变量total是当前感染了感染病的人数,其初始值赋为1,因为节点1为感染源,已经有一个患者了。
(6)变量ans为答案,初始赋值为无穷大,ans=min(ans,total).不断更新最优解。
(7)create_tree(int node)函数用来构建有向树。
(8)dfs(int le) 函数用来层层递归求解
题目代码:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#define INF 0x3f3f3f3f
using namespace std;
vector<int> v[310],t[310];
int level[310];
int vis[310];
int m,n,total,ans;
void create_tree(int node) ///构建有向树
{
vis[node] = 1; //将节点node标记
for(int i = 0; i < v[node].size(); i++) //size函数求出node容器中的元素个数,即与node相邻的节点
{
if(vis[v[node][i]]==0)
{
t[node].push_back(v[node][i]);
create_tree(v[node][i]);
}
}
}
void dfs(int le) //一层一层地向下递归,初始level为1
{
int child = 0; //child变量用来标记level+1层有没有节点,如果有,child=1
if(ans <= total) return; //如果当前情况患病人数比最佳答案ans多或相等,则直接return
for(int i = 1; i <= m; i++) //共有m个人即m个节点,找出在le层的节点
{
if(level[i]==le)
{
for(int j = 0; j < t[i].size(); j++)
{
child = 1; //代表第le+1层上有节点
total++; //将le+1层上的节点数加到total上
level[t[i][j]]=le+1; //节点i的孩子的level值赋为le+1
}
}
}//到此位置total是没有断开情况下,所有的患病人数
if(child==0) ///如果第le+1层没有节点,递归结束条件之1
{
ans = min(ans,total); //获取ans和total的最小值
return;
}
else
{
total--; //每一轮传播都可以选择一个节点与它的父节点断开,则患病人数少1,total减减
for(int i = 1; i <= m; i++)
{
if(level[i]==le+1) //如果节点i在le+1层上,假如切断的节点是i与它前驱间的那条路
{
level[i]=0; ///则将level[i]赋为0,则下次深搜便不会再考虑这个节点以及该节点的子节点。因为level[i]=0!=le+1
dfs(le+1); /// 搜索第le+1层
level[i]=le+1; ///回溯恢复原值
}
}
total++; //回溯,恢复到没有切断路的状态,则被传染的人增加1
for(int i = 1; i <= m; i++) //到此,第le+1层都已经递归求过了。
{
if(level[i]==le+1)
{
level[i]=0; //将节点i的层数赋为0
total--;
}
}
}
}
int main()
{
int i,a,b; ///m个人,n个关系
while(~scanf("%d%d",&m,&n))
{
memset(vis,0,sizeof(vis));
for(i = 1;i <= m; i++)
{
v[i].clear(); //清空容器
t[i].clear();
}
for(i = 1; i <= n; i++)
{
scanf("%d%d",&a,&b);
v[a].push_back(b);
v[b].push_back(a);
}
total = 1;
ans = INF;
create_tree(1) ; //构建有向树
level[1]=1;
dfs(1);
printf("%d\n",ans);
}
return 0;
}
下面举一个例子:测试数据+答案+图示如下:
深搜的结束条件有两个,(1)如果ans<=total,即当前患病的人的数目大于等于最优解ans,则返回。 (2)如果某一层已经没有节点了,ans= min(ans,total) 更新ans后返回。 以给出测试数据为例:下面是我对上图这组测试数据的分析过程,来演示算法的求解过程,以及算法的正确性。 最开始只有level[1]=1
######dfs(1) { child = 0;ans = INF,total = 1; for(i = 1 到 m) { 找到在第一层的点,有节点{1} 则如果不加任何处理,在第一个周期内,则父节点为1的所有子节点将都被传染,则total=1+3=4; 节点1的子节点有2 10 6,因此level[2],level[10],level[6]的值都为2. child = 1因为第二层时有节点的。 } *********此时先判断一下,第二层是否有节点,因为child为1,所以说明第二层有节点,则继续 在每一个周期内,我们一定可以且仅可以选择一个节点与其父节点断开,因此total=4-1=3;则当前实际患病人数为3. 接下来我们开始讨论第2层的节点,谁与父节点断开比较好 for(i = 1 到 m ) { 如果level[i]=2,说明节点i位于第二层上,可以求得有三个节点2,10,6 按算法执行的顺序进行考虑,如果将节点2与1断开,则先将level[2]赋值为0,因为赋值为0后,则执行dfs(2)的时候,就不用考虑2节点后面的子孙节点了。 #####此时执行dfs(2) { child = 0; ans = INF; total = 3; for(i = 1 到 m) { 找到level[i]==2的患病的节点,有节点10和节点6 则total先加上节点10的孩子节点个数,再加上节点6的孩子节点个数,total=9 同时节点10和节点6孩子所在的层次为3,即level[11]=level[12]=level[14]=level[7]= level[8]=level[9]=3; child = 1; } 同理先判断第三层是否有节点,child值为1,代表有。 同理第三层可以选择一个节点与它的父节点断开,total=8,则考虑第三层那个节点与其父节点断开 for(i =1 到 m) { 找到level[i]=3的节点。有11,12,14,7,8,9 按算法顺序递归考虑,则先考虑断开节点11与节点10,则将level[11]赋值为0 #####此时执行dfs(3) { child = 0,ans = INF, total = 8 for(i = 1 到 m) { 找到level[i]=3的患病节点,有12,14,7,8,9 则total加上这些节点的孩子节点数,total=10 此时level[13]=level[15]=4; child = 1; } 同理判断child的值, 同理此轮传播中断开一个节点与其父节点,total=10-1=9 for(i = 1 到 m) { 找到level值为4的节点有,13和15,并按算法顺序进行递归求解,先断开 13和12,则level[13]=0 ######dfs(4) { child = 0,ans = INF, total = 9 for(i = 1 到 m)找到在第四层的节点,有节点15,但是节点15 后面已经没有任何节点了。因此child的值为0, 因此ans = min(ans,total)=min(INF,total)=9,并返回递归上一 层。 } 此时回溯,恢复level[13]的值为4,按算法执行顺序,现在要考虑将15与 它的父节点14断开,因此将level[15]赋值为0,接下来执行递归 #####dfs(4) { child = 0,ans = 9, total = 9 因为ans 和 total相等,则此次递归结束 } 此时回溯,恢复level[15]的值为4,则第四层与第三层断开的情况考虑完 毕 } 恢复total的值,total = 9+1 = 10; 同时将位于第四层的点的level赋为0,并让total减去第四层的节点个数,totol=10-2=8,因为这种情况考虑完了,要进行回溯。 } 接下来又回到for循环了,该考虑12与其父节点10断开的情况了 即level[12]=0 ######dfs(3) { child = 0,ans = 9,total = 8 for(i = 1 到 m) { 找到位于第三层上的节点有11,14,7,8,9。位于这些节点下一层的子节点 只有节点15,因此level[15]=4,total = 8+1 = 9; child = 1; } 同理child为1,代表有子节点 同理可以使第四层中的一个节点与其父节点断开连接 则total = 9-1 = 8 for(i = 1 到 m) { 寻找位于第四层的节点,只有15,让它与父节点断开连接, 则level[15]=0 ######dfs(4) { child = 0,ans = 9,total = 8 for(i = 1 到 m)因为第四层只有15这一个节点,且已断开与上一层的联系 层的连接,因此child=0 则ans = min(ans,total)=min(9,8)=8; 递归返回。 } 循环结束 } total=8+1=9 同时将level[15]赋为0,且total加上第四层的节点数,total=9+1=10; 递归返回上一层 } } ........................接下来依次考虑节点14,7,8,9与上层断开连接的情况,思路与以上分析的一样,可以自行推理。
######dfs(3).....................................
}
}