Codeforces 1086 E. Multihedgehog（dfs）

题目链接
题目大意：
1.1-multihedgehog就是有一个顶点度数大于等于3，其他顶点度数都为1。
2.k-multihedgehog就是有一个顶点数大于3，然后这个节点的所有相邻节点都是一个(k-1)-multihedgehog的中心，然后继续递归下去。
其实画出来就可以看出来要求是一颗至少三叉树（四、五…都可以），树高得是k+1(根节点为1)。
问给定的树是否是一颗至少三叉树。

解题思路：
假设当前树满足条件，则根节点一定会在树的直径的中点，所以先dfs找一下树的直径，然后找到根节点，然后对根节点进行dfs重新记录各个节点深度和各个节点的儿子数量。
最后对每个节点进行判断，如果当前儿子数量为0，就得判断深度是否是最深的叶子的深度，因为所有叶子节点都得在k+1层；如果儿子数量不为0，则判断是否小于3.

解题代码：

#include<bits/stdc++.h>
using namespace std;
mt19937 rng_32(chrono::steady_clock::now().time_since_epoch().count());
typedef long long ll;
const int maxn=1e5+10;
vector<int> v[maxn];
int deg[maxn];
int c[maxn];
bool visit[maxn];
int n,k;
int dep[maxn],pre[maxn],mxdep,dex;
int sz[maxn];
void dfs(int u,int fa,int d)
{
    dep[u]=d;
    if (mxdep<d)
    {
        mxdep=d;
        dex=u;
    }
    for (int i=0;i<v[u].size();i++)
    {
        int to=v[u][i];
        if (to==fa)
        continue;
        sz[u]++;
        pre[to]=u;
        dfs(to,u,d+1);
    }
}
int main()
{
    cin>>n>>k;
    int t1,t2;
    for (int i=1;i<n;i++)
    {
        cin>>t1>>t2;
        v[t1].push_back(t2);
        v[t2].push_back(t1);
        deg[t1]++;
        deg[t2]++;
    }
    int root=0;
    for (int i=1;i<=n;i++)
    {
        if (deg[i]==1)
        {
            root=i;
            break;
        }
    }
    dfs(root,0,1);
    int mid=(1+mxdep)>>1;
    //寻找根节点，在路径上的中点
    while (dex!=root)
    {
        if (dep[dex]==mid)
        {
            root=dex;
            break;
        }
        dex=pre[dex];
    }
    memset(sz,0,sizeof(sz));
    memset(dep,0,sizeof(dep));
    dfs(root,0,1);
    bool f=true;
    for (int i=1;i<=n;i++)
    {
        if(sz[i]==0)
        {
            if (dep[i]!=k+1)
            {
                f=false;
                break;
            }
        }
        else 
        {
            if (sz[i]<3)
            {
                f=false;
                break;
            }
        }
    }
    if (f)
    cout<<"YES\n";
    else
    cout<<"NO\n";
    return 0;
}