Dfs - atlarge

atlarge

描述
给定一个迷宫，构成一棵有根树，你开始在根节点，出口是每个叶子节点，L可以在每个出口放一个守卫，每1个单位时间内，你和守卫都可以移动到相邻的一个点，如果某一时刻 守卫与你相遇了（在边上或点上均算），则你将被抓住。问为了保证抓住你，L最少需要几个守卫。

输入
第1行包含2个用空格分开的正整数n、K，表示有n个节点，K表示根节点编号
接下来n-1行，每行2个整数u,v，表示u到v有条路径

输出
输出1个整数。

样例输入
7 1
1 2
1 3
3 4
3 5
4 6
5 7
样例输出
3
提示
N<=100000

每日牢骚

每天刷着水题……被水题完虐……
全机房都在搞省选+
我还在提高-
哎哎哎……慢慢来吧，咱先弄基础

Analysis

题解很简单
代码也很简单
但我就是想错了

Code

#include<bits/stdc++.h>
#define in read()
#define re register
using namespace std;
inline int read(){
	char ch;int f=1,res=0;
	while((ch=getchar())<'0'||ch>'9') if(ch=='-') f=-1;
	while(ch>='0'&&ch<='9'){
		res=(res<<1)+(res<<3)+(ch^48);
		ch=getchar();
	}
	return f==1?res:-res;
}
const int inf=1e6;
const int N=1e5+10;
const int M=2e5+10;
int n,rt;
int dep[N],mn[N];
int nxt[M],head[N],to[M],ecnt=0;
inline void add(int x,int y){
	nxt[++ecnt]=head[x];head[x]=ecnt;to[ecnt]=y;
}
void dfs1(int u,int fu){
	int son=0;
	for(re int e=head[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fu) continue;
		++son;
		dep[v]=dep[u]+1;
		dfs1(v,u);
		mn[u]=min(mn[u],mn[v]+1);
	}
	if(!son) mn[u]=0;
}
int ans=0;
void getans(int u,int fu){
	if(mn[u]<=dep[u]) {ans++;return;}
	for(re int e=head[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fu) continue;
		getans(v,u);
	} 
}
int main(){
	n=in;rt=in;
	memset(mn,127,sizeof(mn));
	for(re int i=1;i<n;++i){
		int u=in,v=in;
		add(u,v);add(v,u);
	}
	dfs1(rt,0);
	getans(rt,0);
	cout<<ans;
	return 0;
}

错误代码之错误片段

1 判断叶子节点
（肯定不能这样判断啊，应该是看有没有儿子节点）

void dfs1(int u,int fu){
	if(to[head[u]]==fu) mn[u]=0;
	else mn[u]=inf;
	for(re int e=head[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fu) continue;
		dep[v]=dep[u]+1;
		dfs1(v,u);
		mn[u]=min(mn[u],mn[v]+1);
	}
}

2.如果直接走到了儿子节点那就没有统计到了啊

void getans(int u,int fu){
	for(re int e=head[u];e;e=nxt[e]){
		int v=to[e];
		if(v==fu) continue;
		if(mn[v]>dep[v]) getans(v,u);
		else ans++;
	} 
}