本文主要是介绍树与图的深度优先遍历:AcWing 846. 树的重心,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
#include<bits/stdc++.h>
using namespace std;const int N=1e5+10,M=N*2;
int n;
int h[N],e[M],ne[M],idx;
bool state[N];
int ans=N;void add(int a,int b)
{e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}int dfs(int u)
{state[u]=true;int size=0,sum=0;for(int i=h[u];i!=-1;i=ne[i]){int j=e[i];if(state[j]) continue;int s=dfs(j);size=max(size,s);sum+=s;}size=max(size,n-sum-1);ans=min(ans,size);return sum+1;
}int main()
{memset(h,-1,sizeof h);scanf("%d",&n);for(int i=0;i<n-1;i++){int a,b;scanf("%d%d",&a,&b);add(a,b),add(b,a);}dfs(1);printf("%d\n",ans);return 0;
}
单链表的建立,用来表示无向图(操作两次的有向图就是无向图)
size表示的是每一次遍历根节点的连通块的最大值,ans表示的是所有最大值里面的最小值,sum表示的是每一个根节点的子树的节点个数
原理是把某一个点设置为根节点,然后开始搜索,搜索到子节点,然后回溯,对于某一个节点来说,sum表示的是子树的节点个数,我们使用把size和n-sum-1取一个最大值,表示的就是去除当前节点之后,剩余连通块的最大的结点个数,也就是我们需要的答案
dfs里面的循环是遍历链表,递归dfs函数
初始化头节点非常重要
这篇关于树与图的深度优先遍历:AcWing 846. 树的重心的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!