Luogu P1041 传染病控制(搜索)

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;
}

猜你喜欢

转载自www.cnblogs.com/coder-Uranus/p/9903852.html
今日推荐