首页 > 代码库 > luogu cogs . [NOIP2003] 传染病控制 WA(1/2)

luogu cogs . [NOIP2003] 传染病控制 WA(1/2)

★★★   输入文件:epidemic.in   输出文件:epidemic.out   简单对比
时间限制:1 s   内存限制:128 MB

【问题背景】

近来,一种新的传染病肆虐全球。蓬莱国也发现了零星感染者,为防止该病在蓬莱国大范围流行,该国政府 决定不惜一切代价控制传染病的蔓延。不幸的是,由于人们尚未完全认识这种传染病,难以准确判别病毒携带者,更没有研制出疫苗以保护易感人群。于是,蓬莱国 的疾病控制中心决定采取切断传播途径的方法控制疾病传播。经过 WHO (世界卫生组织)以及全球各国科研部门的努力,这种新兴传染病的传播途径和控制方法已经研究消除,剩下的任务就是由你协助蓬莱国疾控中心制定一个有效的控 制办法。

【问题描述】

研究表明,这种传染病的传播具有两种很特殊的性质;第一是它的 传播途径是树型的,一个人 X 只可能被某个特定的人 Y 感染,只要 Y 不得病,或者是 XY 之间的传播途径被切断,则 X 就不会得病。第二是,这种疾病的传播有周期性,在一个疾病传播周期之内,传染病将只会感染一代患者,而不会再传播给下一代。

这 些性质大大减轻了蓬莱国疾病防控的压力,并且他们已经得到了国内部分易感人群的潜在传播途径图(一棵树)。但是,麻烦还没有结束。由于蓬莱国疾控中心人手 不够,同时也缺乏强大的技术,以致他们在一个疾病传播周期内,只能设法切断一条传播途径,而没有被控制的传播途径就会引起更多的易感人群被感染(也就是与 当前已经被感染的人有传播途径相连,且连接途径没有被切断的人群)。当不可能有健康人被感染时,疾病就中止传播。所以,蓬莱国疾控中心要制定出一个切断传 播途径的顺序,以使尽量少的人被感染。

你的程序要针对给定的树,找出合适的切断顺序。

【输入格式】

输入格式的第一行是两个整数 n ( 1≤n≤300 )和 p 。接下来 p 行,每一行有两个整数 i 和 j ,表示节点 i 和 j 间有边相连(意即,第 i 人和第 j 人之间有传播途径相连)。其中节点1 是已经被感染的患者。

【输出格式】

只有一行,输出总共被感染的人数。

【输入样例】

7 6
1 2
1 3
2 4
2 5
3 6
3 7

【输出样例】

3

正解是dfs,而用bfs过5点:

 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue> 
#include<vector>

using namespace std;
const int N=101000;

int answer;
int deep[N];
vector<int> DE[N];
int point,road;
int val[N];
int Max_deep;
queue<int>q;
bool vis[N];
int son[N];
int head[N];
int now=1;
struct node{
    int u,v,nxt;
}E[N];

inline void add(int u,int v)
{
    E[now].v=v;
    E[now].nxt=head[u];
    head[u]=now++;
}

inline int read()
{
    int x=0;char c=getchar();
    while(c<0||c>9)c=getchar();
    while(c>=0&&c<=9)x=x*10+c-0,c=getchar();
    return x;
}

void make_deep()
{
    q.push(1);
    deep[1]=1;
    DE[1].push_back(1);
    while(!q.empty())
    {
        int top=q.front();
        q.pop();
        for(int i=head[top];i!=-1;i=E[i].nxt)
        {
            q.push(E[i].v);
            deep[E[i].v]=deep[top]+1;
            DE[deep[E[i].v]].push_back(E[i].v);
            Max_deep=max(Max_deep,deep[E[i].v]);
        }
    }
}

void make_val()
{
    for(int Q=Max_deep;Q>=1;Q--)
    {
        for(int i=0;i<DE[Q].size();i++)
        {
            int my=DE[Q].size(); 
            int tt=DE[Q][i];
            if(son[tt]==0)
                val[tt]=1;
            else
            {
                for(int j=head[tt];j!=-1;j=E[j].nxt)
                    val[tt]+=val[E[j].v];
                val[tt]++;
            }
        }    
    }
}

inline void change(int start)
{
    q.push(start);
    while(!q.empty())
    {
        int top=q.front();
        q.pop();
        vis[top]=1;
        for(int i=head[top];i!=-1;i=E[i].nxt)
            q.push(E[i].v);
    }
}

inline void find_biggestandwork(int line)
{
    int maxn=-1,maxk;
    for(int i=0;i<DE[line].size();i++)
    {
        int tt=DE[line][i];
        if(val[tt]>maxn&&!vis[tt])
        {
            maxn=val[tt];
            maxk=tt;
        }
    }
    vis[maxk]=1;
    answer-=maxn;
    change(maxk);
}

inline void work()
{
    int Q=Max_deep;
    for(int i=2;i<=Q;i++)
        find_biggestandwork(i);
}

int main()
{
    freopen("epidemic.in","r",stdin);
    freopen("epidemic.out","w",stdout);
    point=read();
    road=read();
    answer=point;
    for(int i=1;i<=point;i++)
        head[i]=-1;
    for(int i=1;i<=road;i++)
    {
        int u=read();
        int v=read();
        son[max(u,v)]++;
        add(min(u,v),max(u,v));
    }
    
    
    make_deep();
    make_val();
    work();
    printf("%d",answer);
    return 0;    
}

/*
7 6
1 2
1 3
2 4
2 5
3 6
3 7
3

*/ 

 

luogu cogs . [NOIP2003] 传染病控制 WA(1/2)