「POJ1655」Balancing Act
Description
Consider a tree T with N (1 <= N <= 20,000) nodes numbered 1…N. Deleting any node from the tree yields a forest: a collection of one or more trees. Define the balance of a node to be the size of the largest tree in the forest T created by deleting that node from T.
For example, consider the tree:
Deleting node 4 yields two trees whose member nodes are {5} and {1,2,3,6,7}. The larger of these two trees has five nodes, thus the balance of node 4 is five. Deleting node 1 yields a forest of three trees of equal size: {2,6}, {3,7}, and {4,5}. Each of these trees has two nodes, so the balance of node 1 is two.For each input tree, calculate the node that has the minimum balance. If multiple nodes have equal balance, output the one with the lowest number.
For example, consider the tree:
Deleting node 4 yields two trees whose member nodes are {5} and {1,2,3,6,7}. The larger of these two trees has five nodes, thus the balance of node 4 is five. Deleting node 1 yields a forest of three trees of equal size: {2,6}, {3,7}, and {4,5}. Each of these trees has two nodes, so the balance of node 1 is two.For each input tree, calculate the node that has the minimum balance. If multiple nodes have equal balance, output the one with the lowest number.
Input
The first line of input contains a single integer t (1 <= t <= 20), the number of test cases. The first line of each test case contains an integer N (1 <= N <= 20,000), the number of congruence. The next N-1 lines each contains two space-separated node numbers that are the endpoints of an edge in the tree. No edge will be listed twice, and all edges will be listed.
Output
For each test case, print a line containing two integers, the number of the node with minimum balance and the balance of that node.
Sample Input
1 2 3 4 5 6 7 8 |
1 7 2 6 1 2 1 4 4 5 3 7 3 1 |
Sample Output
1 |
1 2 |
题意
求树的重心,即选择一个结点删去,使得分出的 若干棵树的结点数 的最大值最小
题解
用size[x]表示x子树的结点数
\[ size_x = \sum size_{y_i} \]
mx[x]表示删去x之后的最大子树结点数
\[ mx_x = max\{size_{y_i},n-size_x\} \]
取mx最小的结点,若有多个,取编号最小的
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
#include<map> #include<set> #include<cmath> #include<stack> #include<queue> #include<cstdio> #include<vector> #include<cstring> #include<cstdlib> #include<iostream> #include<algorithm> #define mod 998244353 #define pi acos(-1) #define inf 0x7fffffff #define ll long long using namespace std; ll read() { ll x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int T,n,ans; int size[20005],mx[20005]; vector<int>e[20005]; void dp(int x,int fa) { size[x]=1; for(int i=0;i<e[x].size();i++) { int y=e[x][i]; if(y==fa)continue; dp(y,x); size[x]+=size[y]; mx[x]=max(mx[x],size[y]); } mx[x]=max(mx[x],n-size[x]); if(mx[x]<mx[ans])ans=x; if(mx[x]==mx[ans]&&x<ans)ans=x; } int main() { T=read(); while(T--) { n=read();ans=0; mx[0]=inf; for(int i=1;i<n;i++) { int u=read(),v=read(); e[u].push_back(v); e[v].push_back(u); } dp(1,0); printf("%d %d\n",ans,mx[ans]); for(int i=1;i<=n;i++)e[i].clear(); memset(size,0,sizeof(size)); memset(mx,0,sizeof(mx)); } return 0; } |
Subscribe