原题链接:蓝桥杯算法提高VIP-传染病控制
解题思路: 使用DFS暴力建树,然后使用DFS暴力搜索加剪枝操作。
参考代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 501;
int n,m,Min=0x3FFFFFFF;
vector<int> box[maxn]; //box[x]保存和x相连的所有的点的编号
int depth[maxn],ill[maxn]; //deptch[x]记录结点x在树中的层次 ,ill[x]记录是否生病,默认为0,表示没有生病
void build_tree(int u,int pre) //求出结点编号为u的结点的深度.pre为u结点的父节点.
{
depth[u]=depth[pre]+1;
for(int i=0;i<box[u].size();i++)
{
int v = box[u][i];
if(v!=pre) build_tree(v,u);
}
}
void dfs(int dep,int sum)
{
if(sum > Min) return ; //剪枝
bool bottom = true; //记录已经到了最后一层了.
for(int i=1;i<=n;i++)
{
if(depth[i]==dep && ill[i])
{
for(int j=0;j<box[i].size();j++)
{
int p = box[i][j];
if(depth[p]==dep+1)
{
bottom = false; //表示dep不是最深的一层.
ill[p]=1;
sum++;
}
}
}
}
sum--;//在同一层中,减去一个分支.
for(int i=1;i<=n;i++)
{
if(depth[i]==dep+1 && ill[i])
{
//注意回溯
ill[i]=0;
dfs(dep+1,sum);
ill[i]=1;
}
}
sum++;
for(int i=1;i<=n;i++)
{
if(depth[i]==dep && ill[i])
{
for(int j=0;j<box[i].size();j++)
{
int p = box[i][j];
if(depth[p]==dep+1)
{
ill[p]=0;
sum--;
}
}
}
}
if(bottom) Min=min(Min,sum);
}
int main(void)
{
//freopen("D:\\input.txt","r",stdin);
cin >> n >> m;
for(int i=0;i<m;i++)
{
int u,v;
cin >> u >> v;
box[u].push_back(v);
box[v].push_back(u);
}
build_tree(1,0);
ill[1]=1;
dfs(1,1);
cout << Min ;
return 0;
}0.0分
0 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复