[蓝桥杯][算法提高VIP]传染病控制

题目

题目链接

题解

DFS。

---

大致思路：
按层深搜，遍历每一层中被感染的，将其直接子节点全部设置为被感染，统计被感染的数目，每次深搜的时候，我们只需要选定其中一个直接子节点作为截断处，将此结点设置为未被感染，进行深搜。回溯的时候再将标记再恢复为感染，恢复被感染的数目。当到达最深层时，更新最优解。

注意剪枝，要不超时。

---

有点难，虽然看题解看懂了，但是感觉理解的不透彻。

代码

#include<bits/stdc++.h>
using namespace std;
const int N = 310;

int a, b, n, p, ans = N, ill[N], fa[N];
vector<int> v[N], dp[N];

// 构建dp和fa 
void getdp(int x, int f, int d) { // x：点编号  f：x的父节点编号  d：深度 
	dp[d].push_back(x); // 深度为d的结点中包含x 
	fa[x] = f; // x的父亲是f 
	for(int i = 0;i < v[x].size();i ++) {
		int j = v[x][i];
		if(j == f) continue;
		getdp(j, x, d+1);
	}
}

void dfs(int d, int sum) { // d：深度   sum：感染数 
	
	if(ans < sum) return ; // 剪枝 
	
	int isbottom = 1; // 是否到达最深层 
	for(int i = 0;i < dp[d].size();i ++) {
		int t = dp[d][i];
		if(ill[t]) { // 找到d层中被感染的点有哪些，将这些点的孩子结点也全部感染 
			for(int j = 0;j < v[t].size();j ++) {
				int s = v[t][j];
				if(fa[t] == s) continue;
				isbottom = 0; // 未到最深层 
				ill[s] = 1; // 标记为被感染了 
				sum ++; // 感染数 
			}
		}
	}
	
	sum --; // 相当于从上面全部的孩子结点中，选出一个不被感染的，作为截断路径两端结点中深层的结点 
	
	for(int i = 0;i < dp[d+1].size();i ++) { // 遍历d+1层的结点 
		int t = dp[d+1][i];
		if(ill[t]) { // 若被感染，则设置为未被感染，相当于这就是咱们选定的截断路径 
			ill[t] = 0;
			dfs(d+1, sum);
			ill[t] = 1; // 恢复，再试着选下一个 
		}
	}
	
	sum ++; // 再标记回去，还是孩子节点全部感染 
	 
	for(int i = 0;i < dp[d].size();i ++) { // 这部分循环与最上面的循环一致，但是目的相反，这里是将全部孩子结点恢复为未被感染 
		int t = dp[d][i];
		if(ill[t]) {
			for(int j = 0;j < v[t].size();j ++) {
				int s = v[t][j];
				if(fa[t] == s) continue;
				ill[s] = 0;
				sum --;
			}
		}
	}
	
	if(isbottom) ans = min(ans, sum);
}

int main() {
	cin>>n>>p;
	while(p--) {
		cin>>a>>b;
		v[a].push_back(b);
		v[b].push_back(a);
	}
	
	getdp(1, 0, 1); 
	
	ill[1] = 1;
	dfs(1, 1);
	
	cout << ans << endl;
	
	return 0;
}