问题描述:
研究表明,这种传染病的传播具有两种很特殊的性质;
第一是它的传播途径是树型的,一个人X只可能被某个特定的人Y感染,只要Y不得病,或者是XY之间的传播途径被切断,则X就不会得病。
第二是,这种疾病的传播有周期性,在一个疾病传播周期之内,传染病将只会感染一代患者,而不会再传播给下一代。
这些性质大大减轻了蓬莱国疾病防控的压力,并且他们已经得到了国内部分易感人群的潜在传播途径图(一棵树)。但是,麻烦还没有结束。由于蓬莱国疾控中 心人手不够,同时也缺乏强大的技术,以致他们在一个疾病传播周期内,只能设法切断一条传播途径,而没有被控制的传播途径就会引起更多的易感人群被感染(也 就是与当前已经被感染的人有传播途径相连,且连接途径没有被切断的人群)。当不可能有健康人被感染时,疾病就中止传播。所以,蓬莱国疾控中心要制定出一个 切断传播途径的顺序,以使尽量少的人被感染。你的程序要针对给定的树,找出合适的切断顺序。
输入:
输入格式的第一行是两个整数n(1≤n≤300)和p。
接下来p行,每一行有两个整数i和j,表示节点i和j间有边相连(意即,第i人和第j人之间有传播途径相连,注意:可能是i到j也可能是j到i)。其中节点1是已经被感染的患者。
对于给定的输入数据,如果不切断任何传播途径,则所有人都会感染。
输出:
只有一行,输出总共被感染的人数。
样例输入:
7 6
1 2
1 3
2 4
2 5
3 6
7 3
样例输出:
3
思路:
这题跟树有关,所以优先想到DFS,但是很容易超时,这里是引用一个大佬(列公子)的观念
1. 问题转化
所有结点构成一棵树,在树的每一层摘除一个节点(及其子节点),使得保留的树, 其节点数最小,即感染的人数最少。
也就是尽可能多地 去掉树中的节点。
2. 数据结构
树采用邻接表存储, 即 vector<int> Adj[maxn], 使用邻接矩阵 超时 18 %
用一个类似邻接表的结构, 存储树中每一层有哪些节点 (存储其编号), 即 vec<int> depths
布尔型数组 bool isRemoved[maxn] 标记对应编号的节点是否已被摘除。
3. 算法
initDepths用来初始化 depths, 即递归地求每个节点的深度。
changeState 用来摘除节点(及其子节点) 或 恢复节点(及其子节点), 通过递归地操作 isRemoved 数组实现, 同时统计该子树的节点总数。
dfs 深度优先地搜索 树的每一层
#include <bits/stdc++.h>
using namespace std;
const int maxn = 310;
int N, P