数组方式存储树/图-以“树的重心”一题为例

给定一颗树，树中包含 n𝑛 个结点（编号 1∼n1∼𝑛）和 n−1𝑛−1 条无向边。

请你找到树的重心，并输出将重心删除后，剩余各个连通块中点数的最大值。

重心定义：重心是指树中的一个结点，如果将这个点删除后，剩余各个连通块中点数的最大值最小，那么这个节点被称为树的重心。

输入格式

第一行包含整数 n𝑛，表示树的结点数。

接下来 n−1𝑛−1 行，每行包含两个整数 a𝑎 和 b𝑏，表示点 a𝑎 和点 b𝑏 之间存在一条边。

输出格式

输出一个整数 m𝑚，表示将重心删除后，剩余各个连通块中点数的最大值。

数据范围

1≤n≤105

输入样例

9
1 2
1 7
1 4
2 8
2 5
4 3
3 9
4 6

输出样例：

4

#include<iostream>
using namespace std;
#include<algorithm>
#include<cstring>

#define INT_MAX 0x7777777;

//这道题的题面是树，所以用指针不方便存储树结构，故用数组
//h[u] = idx ：h中存储的是每个结点邻接点的idx，用结点值进行索引
//e[idx] = u ：e中存储的是每个结点的值，用idx进行索引
//ne[idx] = idx ：ne中存储的是每个结点的idx，用idx进行索引

//这种方式与struct指针方式存储的区别在于遍历的方式
//指针遍历只需要 dfs(root->left)
//数组遍历需要 for(int i = h[u]; i != -1; i = ne[i])
//相比于前者，后者在速度上更快，也更加灵活，但是需要一定时间领悟

const int N = 200020;
int h[N], e[N], ne[N], idx;
int ans = INT_MAX;
int n;

bool st[N];//状态数组

//在a和b之间添加一个连线
void add(int a, int b){
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

int dfs(int u){
    //搜索以u为根的子树的结点值
    //去掉u之后的每个联通量的最大值 = max(u的每个子节点为根的子树的结点值, 总结点数 - u的每个子节点为根的子树的结点值 - 1)
    //所以应该先搜索到底部
    st[u] = true;
    int size = 0, sum = 0;//每个子树结点值的最大值，各个子树的结点值总和
    for(int v = h[u]; v != -1; v = ne[v]){
        int j = e[v];
        if(st[j])continue; //因为数组存储树的方式会产生双向边，因此需要一个状态数组放置重复遍历

        int s = dfs(j); //以j为根节点的子树的大小
        size = max(size, s);
        sum += s;
    }

    size = max(n - sum - 1, size);
    ans = min(ans, size);

    return sum + 1;
}

int main(){
    memset(h, -1, sizeof h);
    cin>>n;

    for(int i = 0;i < n - 1;i++){
        int a, b;
        cin>>a>>b;
        add(a, b);
        add(b, a);
    }

    dfs(1);
    cout<<ans;

    return 0;
}