POJ 1655 Balancing Act

这道题要求删除某点之后最大子树大小最小，也就是求树的重心。

把一个结点当做根之后，它的最大子树大小可以这样计算：

按照任意一个点作为根建树之后 dfs ，把原树转成有根树（这里用了 1 号节点）；然后对于一个结点 o，计算转换后的o的最大子树的大小，和把 o 作为根之后，通过现在父亲一侧形成的子树的大小，两者取最大值就是 o 作为根时的最大子树大小。在 dfs 的同时记录重心即可。

注意如果是用前向星构图，每条边要建两次，所以数组要开两倍。（然后我就因此RE了……）
代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 40000 + 10//注意数组大小！
#define isdgt(x) (x >= '0' && x <= '9')

using namespace std;

inline void read(int& x) {
    x = 0;
    bool f = true;
    char ch = getchar();
    while(!isdgt(ch)) f = ch == '-' ? false : f, ch = getchar();
    while(isdgt(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar();
    x = f ? x : -x;
}

int n;
int edcnt;
int star[maxn];
int to[maxn], nxt[maxn];
int fa[maxn], siz[maxn], grav, size;
//siz数组是结点的子节点总数，不包括自己
//grav是重心，size是重心的最大子树大小

void init() {
    size = maxn;
    edcnt = grav = 0;
    memset(fa, 0, sizeof(fa));
    memset(to, 0, sizeof(to));
    memset(nxt, 0, sizeof(nxt));
    memset(siz, 0, sizeof(siz));
    memset(star, -1, sizeof(star));
}

void addsedge(int f, int t) {
    to[edcnt] = t;
    nxt[edcnt] = star[f];
    star[f] = edcnt++;
}

void addedge(int u, int v) {
    addsedge(u, v);
    addsedge(v, u);
}

void dfs(int o) {
    int maxsize = 0;
    for(int i = star[o]; i != -1; i = nxt[i]) {
        if(to[i] == fa[o]) continue;
        fa[to[i]] = o; dfs(to[i]);
        int advance = siz[to[i]] + 1;//刚访问的子树大小（加1是因为siz[to[i]]中没有算上to[i]这个点
        maxsize = max(maxsize, advance);
        siz[o] += advance;
    }
    maxsize = max(maxsize, n - siz[o] - 1);//n - siz[o] - 1就是通过父亲形成的子树大小
    if(maxsize < size) {
        grav = o, size = maxsize;
    } else if(maxsize == size) {
        grav = grav > o ? o : grav;
    }
}

int main() {
    int casenum;
    read(casenum);
    for(int caseno = 0; caseno < casenum; caseno++) {
        init(), read(n);
        for(int i = 1; i < n ; i++) {
            int u, v;
            read(u), read(v);
            addedge(u, v);
        }
        dfs(1);
        printf("%d %d\n", grav, size);
    }
    return 0;
}

评测结果：

Result	Memory	Time	Language	Code Length
Accepted	944K	110MS	C++	1597B

（RE了一次……数组一定不要开小了……）