P1041 [NOIP2003 提高组] 传染病控制（dfs暴力）

P1041 [NOIP2003 提高组] 传染病控制（dfs）

每一层断掉一条边
使得剩余最小

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node{
	ll u,v,next;
}e[100000];
int cnt=0;
ll head[100000],siz[10000],ans,dep[100000],fa[100000],vis[100000];
ll n,m;
vector<int>k[10000];
void add(int u,int v){
	e[++cnt]=(node){u,v,head[u]};
	head[u]=cnt;
}
int maxdep=-1;
void dfs(int x,int f,int deep){

	fa[x]=f;siz[x]=1;dep[x]=deep;
	maxdep=max(maxdep,deep);
	for(int i=head[x];i;i=e[i].next)
		if(e[i].v!=f){
			dfs(e[i].v,x,deep+1);
			siz[x]+=siz[e[i].v];
		}
}

void dffs(ll dep,ll now){
	if(dep==maxdep+1){
		ans=min(ans,now);
		return;
	}
	
	for(auto i:k[dep]){
		if(vis[fa[i]])
		vis[i]=1;
		else
		vis[i]=0;
	}
	int flag=1;
	for(auto i:k[dep]) if(!vis[i]) flag=0;
	if(flag){
		ans=min(ans,now);
		 return;
	}
	
	for(auto i:k[dep]){
		
		if(!vis[i]){
			
			vis[i]=1;
			dffs(dep+1,now-siz[i]);
			vis[i]=0;
		}
	}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		int x,y;
		cin>>x>>y;
		add(x,y); add(y,x);
	}
	dfs(1,0,1);
	for(int i=1;i<=n;i++){
		k[dep[i]].push_back(i);
	}
	ans=n;
	dffs(2,n);
	cout<<ans;
	return 0;
}