树与图的深度优先遍历:AcWing 846. 树的重心

#include
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

 单链表的建立,用来表示无向图(操作两次的有向图就是无向图)

size表示的是每一次遍历根节点的连通块的最大值,ans表示的是所有最大值里面的最小值,sum表示的是每一个根节点的子树的节点个数

原理是把某一个点设置为根节点,然后开始搜索,搜索到子节点,然后回溯,对于某一个节点来说,sum表示的是子树的节点个数,我们使用把size和n-sum-1取一个最大值,表示的就是去除当前节点之后,剩余连通块的最大的结点个数,也就是我们需要的答案

dfs里面的循环是遍历链表,递归dfs函数

初始化头节点非常重要

你可能感兴趣的:(算法竞赛,深度优先,算法,图论)