2018年全国多校算法寒假训练营练习比赛(第二场)E
只写了四题,220名开外,,,菜到泣不成声
[题目链接](https://www.nowcoder.com/acm/contest/74/E)
题目描述
小G想要把自己家院子里的橘子树搬到家门口(QAQ。。就当小G是大力水手吧)
可是小G是个平衡性灰常灰常差的人,他想找到一个这个橘子树的平衡点。
怎么描述这棵树呢。。。就把它看成由一个个节点构成的树吧。结点数就
代表树重。
输入描述:
多组数据输入输出,
第一行包含一个整数n(3<=n<=1000)代表树的结点的个数
以下n-1行描述(1-n)节点间的连接关系。
输出描述:
输出两个个整数 x,num 分别代表树的平衡点,和删除平衡点后最大子树的结点数(如果结点数相同输出编号小的)。
示例1
输入
3
1 2
1 3
输出
1 1
解题思路
树分治模板中的找重心模板,非常猥琐的套了一波模板-_-!!!
AC代码
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
const int maxn = 10010;
int N, K;
int ans, root, Max;
struct node
{
int v, next;
}edge[maxn * 2];
int head[maxn], tot;
int Size[maxn];//树的大小
int maxv[maxn];//最大孩子节点的size
int vis[maxn];
int dis[maxn];
int num;
void init()
{
tot = 0;
ans = 0;
memset(head, -1, sizeof(head));
memset(vis, 0, sizeof(vis));
}
void add_edge(int u, int v)
{
edge[tot].v = v;
edge[tot].next = head[u];
head[u] = tot++;
}
//处理子树的大小
void dfssize(int u, int f)
{
Size[u] = 1;
maxv[u] = 0;
for (int i = head[u]; i != -1; i = edge[i].next)
{
int v = edge[i].v;
if (v == f || vis[v])continue;
dfssize(v, u);
Size[u] += Size[v];
if (Size[v]>maxv[u])maxv[u] = Size[v];
}
}
//找重心
void dfsroot(int r, int u, int f)
{
if (Size[r] - Size[u]>maxv[u])//size[r]-size[u]是u上面部分的树的尺寸,跟u的最大孩子比,找到最大孩子的最小差值节点
maxv[u] = Size[r] - Size[u];
if (maxv[u]<Max)Max = maxv[u], root = u;
for (int i = head[u]; i != -1; i = edge[i].next)
{
int v = edge[i].v;
if (v == f || vis[v])continue;
dfsroot(r, v, u);
}
}
void dfs(int u)
{
Max = N;
dfssize(u, 0);
dfsroot(u, u, 0);
//ans += calc(root, 0);
}
int main()
{
while (scanf("%d", &N) != EOF)
{
if (!N)break;
int u, v;
init();
for (int i = 1; i<N; i++)
{
scanf("%d%d", &u, &v);
add_edge(u, v);
add_edge(v, u);
}
Max = N;
dfssize(1, 0);
dfsroot(1, 1, 0);
printf("%d %d\n", root,Max);
}
return 0;
}