CodeForces - 1182D

题意：找到一个点作根，使同一深度的点有相同数目的边。

 

思路：经过简单的观察，实际上这样的树有一个特征：当某一个点作根符合题目要求时，每个相同深度的节点其子树的节点数目是相同的，或者是每个相同深度的节点作根，其子树结构相同。

那么答案只有两种情况，一种是叶子，一种是某个点（唯一一个），其每一个所连接的子树的节点数目相等。

因此面对判断一个点是否是答案的难题，可以比较每个节点作根时其子树节点的数目。

虽然做法很暴力，但是可以通过一系列的剪枝操作优化。

如果是判断叶子，一般运行到第一个出现两个以上条边的特殊节点就可以知道是否正确，除非此时答案属于第二种。

因为多个叶子作根可行，其子树大小要求是定死的，而且共享上述所提到特殊节点。

 

至于第二种只有唯一一个，可以使用反证法。如果存在这么一个点，那么另外一个点存在于它的子树中，然而此时就会出现一颗节点数目小于其他树的子树。

 

#include <bits/stdc++.h>
using namespace std;

#define LL long long
#define MAX 100010
#define MOD 998244353

vector<int> e[MAX];
vector<int> a;

int n;
int siz[MAX];
int id[MAX];
int d[MAX];
int c[MAX];
int f[MAX];
int dfn;


void dfs(int x,int fa,int *y){

    id[x]=++dfn;
    siz[x]++;
    for(int i=0;i<e[x].size();i++){
        if(e[x][i]==fa)continue;
        dfs(e[x][i],x,y);
        siz[x]+=siz[e[x][i]];
    }
    if(*y>0)
		return ;
    int temp=(n-siz[x]);
    bool flag=true;
    for(int i=0;i<e[x].size();i++){
        if(e[x][i]==fa)continue;
        if(temp!=siz[e[x][i]]){
            flag=false;
        }
    }
    if(flag&&siz[x]>1){
        *y=x;
    }
}

bool judge(int x,int fa,int dep){
	
	if(c[x]!=0&&f[x]!=fa&&c[x]!=fa)
		return false;
	f[x]=0;
    for(int i=0;i<e[x].size();i++){
        int nt=e[x][i];
        if(nt==fa)continue;
        
        int son_siz=siz[nt];
        if(id[x]>id[nt])
            son_siz=n-siz[x];
        if(d[dep]==0){
            d[dep]=son_siz;
        }
        if(d[dep]!=son_siz){
			c[x]=nt;
            return false;
        }
        f[x]=nt;
    }
    
    for(int i=0;i<e[x].size();i++){
        int nt=e[x][i];
        if(nt==fa)continue;
        if(!judge(nt,x,dep+1)){
            c[x]=nt;
            return false;
        }
		f[x]=nt;
    }
    f[x]=0;
    return true;
}

inline bool scan_d(int &num)  
{
	char in;bool IsN=false;
	in=getchar();
	if(in==EOF) return false;
	while(in!='-'&&(in<'0'||in>'9')) in=getchar();
	if(in=='-'){ IsN=true;num=0;}
	else num=in-'0';
	while(in=getchar(),in>='0'&&in<='9'){
			num*=10,num+=in-'0';
	}
	if(IsN) num=-num;
	return true;
}

int main() {
	
    int x=-1,u,v;
    scanf("%d",&n);
    
    for(int i=1;i<n;i++){
        scan_d(u);
        scan_d(v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    
    dfs(1,1,&x);
    
    for(int i=1;i<=n;i++){
        if(e[i].size()<=1){
            a.push_back(i);
        }
    }
    
    if(x!=-1){
        a.push_back(x);
    }

    for(int i=0;i<a.size();i++){
    	
        if(judge(a[i],a[i],1)){
            printf("%d\n",a[i]);
            return 0;
        }

        for(int i=1;;i++){
            if(d[i]==0)break;
            d[i]=0;
        }
    }
    printf("-1\n");
    return 0;
}