cf 566 D. Complete Mirror 图

D. Complete Mirror

题目链接

题目大意

给出一棵树，让你找一个根节点，使每个到根节点距离相等的点的度数都相等。
怎么做呢？ 想一下答案有几种情况：
1、所有叶子节点。这个应该都可以想到
不是叶子结点的情况：

又是歪歪又扭扭的图
这个图的答案只能是4。
所以：
2、树的重心。这他妈谁想得到，知道有这种情况 没想到这这是个重心
知道这个以后就简单了：
先判断重心，然后枚举叶子节点、就好了。
但是枚举全部叶子节点肯定超时（菊花图）。
然后想一想。距离重心距离相等的叶子节点只用记一个就好。如果这个可以那么别的肯定也可以，这些是等价的。
所以就用vector存叶子节点就好。
所以就简单了。
代码：

#include <stdio.h>
#include <algorithm>
#include <vector>
#include <queue>
#include <string.h>
using namespace std;
const int maxn = 1e5+5;
queue<int> qq;
int rd[maxn];
int cd[maxn];
vector<int> vv[maxn];
int num[maxn];
vector<int> ss;

void dfs1(int x,int fa,int dp)
{
    if(num[dp] == 0 && cd[x] == 1)
    {
        num[dp] = 1;
        ss.push_back(x);
    }
    for (int i = 0; i < vv[x].size(); i ++ )
    {
        int v = vv[x][i];
        if(v == fa)
            continue;
        if(cd[v] == 2 || cd[v] == 1)
            dfs1(v,x,dp + 1);
    }
}



int tp()
{
    int ans = 1;
    while(!qq.empty())
    {
        int x = qq.front();
        qq.pop();
        ans = x;
        for (int i = 0; i < vv[x].size(); i ++ )
        {
            int v = vv[x][i];
            rd[v] -- ;
            if(rd[v] == 1)
            {
                qq.push(v);
                rd[v] = 0;
            }
        }
    }
    return ans;
}
int f = 1;

void dfs2(int x,int fa,int dep)
{
    if(f == 0)
        return;
    if(num[dep] == 0)
        num[dep] = cd[x];
    else if(num[dep] != cd[x])
    {
        f = 0;
        return;
    }
    for (int i = 0; i < vv[x].size(); i ++ )
    {
        int v = vv[x][i];
        if(v == fa)
            continue;
        dfs2(v,x,dep+1);
    }
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i = 1; i < n; i ++ )
    {
        int x,y;
        scanf("%d%d",&x,&y);
        vv[x].push_back(y);
        vv[y].push_back(x);
        cd[x] ++ ;
        cd[y] ++ ;
        rd[x] ++ ;
        rd[y] ++ ;
    }
    for (int i = 1; i <= n; i ++ )
    {
        if(rd[i] == 1)
        {
            rd[i] = 0;
            qq.push(i);
        }
    }
    int root = tp();
    dfs1(root,0,1);
    ss.push_back(root);
    for (int i = 0; i < ss.size(); i ++ )
    {
        f = 1;
        memset(num,0,sizeof(num));
//        printf("%d\n",ss[i]);
        dfs2(ss[i],0,1);
        if(f)
        {
            printf("%d\n",ss[i]);
            return 0;
        }
    }
    printf("-1\n");
}