【NOIP2003 提高组】传染病控制

题目

https://www.luogu.org/problemnew/show/P1041

思路

题目大意是：把一棵树按深度分层，每一层断掉一条边，是剩下的节点数最小。

其实，我们可以将问题转换为断掉的节点数最多。

首先，贪心不可行，很容易被卡。

因为数据只有300，直接搜索就行。

搜索时一层一层搜，枚举断掉哪条边，并标记后代。

代码

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int maxn=377;
int n,p,f[maxn],s=0,ans=0;
vector<int> d[maxn],a[maxn];
void init(int u,int dep)
{
    d[dep].push_back(u);
    for(int i=0; i<a[u].size(); i++)
    {
        init(a[u][i],dep+1);
    }
}
int work(int u,int t)
{
    int ss=1; f[u]=t;
    for(int i=0; i<a[u].size(); i++)
    {
        ss+=work(a[u][i],t);
    }
    return ss;
}
void dfs(int dep)
{
    for(int i=0; i<d[dep].size(); i++)
    {
        if(f[d[dep][i]]) continue;
        s+=work(d[dep][i],1); ans=max(ans,s);
        dfs(dep+1); 
        s-=work(d[dep][i],0);
    }
}
int main()
{
    scanf("%d%d",&n,&p);
    for(int i=1; i<=p; i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        if(x>y) swap(x,y);
        a[x].push_back(y);
    }
    init(1,1);
    dfs(2);
    printf("%d",n-ans);
}