基础算法 846.树的重心(数组模拟邻接表)


用一个数组h[ ]储存图中的节点。

每个节点引出一个链表,表明从这个点连接出去的边。

类似 模拟散列表中的拉链法。

由于树是一个无权图。

所以在建立树时使用add(a, b)和add(b, a)。

ne[ ]表示一个顶点指向的下一个顶点的指针。

e[ ]表示当前节点idx,点的序号。

add(int a, int b){
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
#include
#include
using namespace std;

const int N = 100010, M = N * 2;
int h[N], ne[M], e[M], ans = N, idx, n;
bool state[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 sum = 1, res = 0;
    for(int i = h[u]; i != -1; i = ne[i]){
        
        int j = e[i];
        
        if(!state[j]){
            int s = dfs(j);
            res = max(res, s);
            sum += s;
        }
    }
    res = max(res, n - sum);
    ans = min(res, ans);
    return sum;
}

int main(){
    cin >> n;
    memset(h, -1, sizeof h);
    
    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;
}

相关