POJ 1655 Balancing Act (dfs)

Balancing Act

上英语课时想到一个方法，用map+vector记录数据，然后以每一个节点为根节点搜一次。回去一试tle了，也难怪。后来发现数据1-n连续的，就不用map了，直接数组就可以了。

想来想去，发现一种方法，以1为根，先用一个dfs把每个子树的结点树记录，然后再搜一遍找最小值。本来不难，因为数据输入弄错了，调了好长时间，真郁闷。

 

#include <iostream> #include <cstring> #include <cmath> #include <cstdlib> #include <algorithm> #include <string> #include <cstdio> #include <climits> #include <queue> #include <map> #include <list> #include <vector> using namespace std; vector<int> mm[20010]; int summ[20010]; int n, minn, res; int solve(int b, int a = 0) { if (a == 0) { summ[b] = n; for (unsigned i=0; i < mm[b].size(); i++) { solve(mm[b][i], b); } return 0; } else { summ[b] = 1; for (unsigned i=0; i < mm[b].size(); i++) { if (mm[b][i] != a) summ[b] += solve(mm[b][i], b); } return summ[b]; } } void dfs(int b, int a = 0) { int maxn = n - summ[b]; if (maxn >= minn) return; for (unsigned i=0; i < mm[b].size(); i++) { if (mm[b][i] != a && maxn < summ[mm[b][i]]) maxn = summ[mm[b][i]]; } if (minn > maxn) { minn = maxn; res = b; } for (unsigned i=0; i < mm[b].size(); i++) { if (mm[b][i] != a) dfs(mm[b][i], b); } } int main() { int ti; scanf("%d", &ti); while (ti--) { scanf("%d", &n); for (int i=1; i <=n; i++) mm[i].clear(); for (int i=1; i < n; i++) { int a, b; scanf("%d%d", &a, &b); mm[a].push_back(b); mm[b].push_back(a); } minn = 0x7fffffff; solve(1); dfs(1); printf("%d %d/n", res, minn); } return 0; }