acwing 846 树的重心

dfs暴力搜索

思路

这个题目就是题目意思有点绕，但是用一个样例就可以理解其中意思。

主要是dfs的搜索过程

我们用dfs返回，表示以u为节点的其中一个连通块的数量:sum，另外一个我们怎么表示？其实画画图就知道了，就是n-sum。那么我们每次得到一个sum和n-sum。我们就取一次最大值（题目意思）。然后在该点完成后，我们再和最终的结果取一个min。直到所有的点都访问过了，我们就可以得到最后的答案。

代码

有个细节需要注意一下，那就是是无向边然后就是e,ne,数组需要开两倍，不然就会有样例过不去。

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
const int N = 100010,M = 2*N;
int h[N],ne[M],e[M],idx=0;
bool st[N];
int ans = N;
void add(int a,int b){
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}
//以u为根中点的子树的数量
int dfs(int u){
    int sum = 1,res = 0;
    st[u] = true;
    for(int i = h[u];i!=-1;i=ne[i]){
        int j = e[i];
        if(!st[j]){
            //子树的数量
            int s = dfs(j);
            res = max(s,res);
            sum+=s;
        }
    }
    res = max(res,n-sum);
    ans = min(ans,res);
    return sum;//返回的是子树的点数
}

int main()
{
    scanf("%d",&n);
    memset(h,-1,sizeof h);
    for(int i =0;i<n-1;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);//无向边要加两次
    }
    dfs(1);
    printf("%d\n",ans);
    return 0;
}