Luogu P1041 传染病控制(搜索)
题意
题目背景
近来,一种新的传染病肆虐全球。蓬莱国也发现了零星感染者,为防止该病在蓬莱国大范围流行,该国政府决定不惜一切代价控制传染病的蔓延。不幸的是,由于人们尚未完全认识这种传染病,难以准确判别病毒携带者,更没有研制出疫苗以保护易感人群。于是,蓬莱国的疾病控制中心决定采取切断传播途径的方法控制疾病传播。经过\(WHO\)(世界卫生组织)以及全球各国科研部门的努力,这种新兴传染病的传播途径和控制方法已经研究清楚,剩下的任务就是由你协助蓬莱国疾控中心制定一个有效的控制办法。
题目描述
研究表明,这种传染病的传播具有两种很特殊的性质;
第一是它的传播途径是树型的,一个人\(X\)只可能被某个特定的人\(Y\)感染,只要\(Y\)不得病,或者是\(XY\)之间的传播途径被切断,则\(X\)就不会得病。
第二是,这种疾病的传播有周期性,在一个疾病传播周期之内,传染病将只会感染一代患者,而不会再传播给下一代。
这些性质大大减轻了蓬莱国疾病防控的压力,并且他们已经得到了国内部分易感人群的潜在传播途径图(一棵树)。但是,麻烦还没有结束。由于蓬莱国疾控中心人手不够,同时也缺乏强大的技术,以致他们在一个疾病传播周期内,只能设法切断一条传播途径,而没有被控制的传播途径就会引起更多的易感人群被感染(也就是与当前已经被感染的人有传播途径相连,且连接途径没有被切断的人群)。当不可能有健康人被感染时,疾病就中止传播。所以,蓬莱国疾控中心要制定出一个切断传播途径的顺序,以使尽量少的人被感染。
你的程序要针对给定的树,找出合适的切断顺序。
输入输出格式
输入格式:
第一行是两个整数\(n(1\leq n\leq 300)\)和\(p\)。
接下来\(p\)行,每一行有\(2\)个整数\(i\)和\(j\),表示节点\(i\)和\(j\)间有边相连。(意即,第\(i\)人和第\(j\)人之间有传播途径相连)。其中节点\(1\)是已经被感染的患者。
输出格式:
\(1\)行,总共被感染的人数。
输入输出样例
输入样例#1:
7 6
1 2
1 3
2 4
2 5
3 6
3 7
输出样例#1:
3
思路
暴搜,然后来慢慢优化算法。
暴搜的方法也很简单,枚举每次切断的途径,再来判断接下来传染病如何传播,模拟即可。
接下来考虑优化。首先有个贪心的想法:切断当前刚被感染的点,这样一定是最优的切法,因为这样可以保护更多的子树结点。
如果我们不去切树,会发现点被感染的时间其实就是点在树中的深度。所以我们可以先预处理各点的深度,每次就可以直接得到新感染的点有哪些。
再加上最优性剪枝,即如果还没有感染的点已经小于之前搜索得到的答案了,直接返回。其实这么剪枝已经可以\(AC\)了,但是实际上如果加上记忆化剪枝,可以大大加快运行速度,不过这个我就没有尝试了。
AC代码
#include<bits/stdc++.h>
using namespace std;
const int MAXN=305;
const int MAXM=10000005;
int n,m,ans=INT_MAX,mxdep,dep[MAXN],sz[MAXN];
int cnt,top[MAXN],to[MAXN<<1],nex[MAXN<<1];
int _cnt,_top[MAXN],_to[MAXM],_nex[MAXM];
int read()
{
int re=0;char ch=getchar();
while(!isdigit(ch)) ch=getchar();
while(isdigit(ch)) re=(re<<3)+(re<<1)+ch-'0',ch=getchar();
return re;
}
void dfs1(int now)
{
mxdep=max(mxdep,dep[now]);
for(int i=top[now];i;i=nex[i])
{
if(sz[to[i]]) continue;
dep[to[i]]=dep[now]+1,sz[to[i]]=1;
dfs1(to[i]);
sz[now]+=sz[to[i]];
}
}
void dfs2(int dp,int rest,int cut)
{
if(dp==mxdep)
{
ans=min(ans,rest);
return ;
}
int tot=0;_top[dp+1]=0;
for(int i=_top[dp];i;i=_nex[i])
{
if(_to[i]==cut) continue;
for(int j=top[_to[i]];j;j=nex[j])
if(dep[to[j]]==dp+1) tot+=sz[to[j]],_to[++_cnt]=to[j],_nex[_cnt]=_top[dp+1],_top[dp+1]=_cnt;
}
if(rest-tot>=ans) return ;
if(!tot)
{
ans=min(ans,rest);
return ;
}
for(int i=_top[dp+1];i;i=_nex[i]) dfs2(dp+1,rest-sz[_to[i]],_to[i]);
}
int main()
{
n=read(),m=read();
while(m--)
{
int x=read(),y=read();
to[++cnt]=y,nex[cnt]=top[x],top[x]=cnt;
to[++cnt]=x,nex[cnt]=top[y],top[y]=cnt;
}
dep[1]=sz[1]=1;
dfs1(1);
_to[++_cnt]=1,_nex[_cnt]=_top[1],_top[1]=_cnt;
dfs2(1,n,0);
printf("%d",ans);
return 0;
}