【模拟】Codeforces - 1067B - Multihedgehog

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/monochrome00/article/details/83830675

---

题目链接<http://codeforces.com/contest/1067/problem/B>

---

题意：

定义一张棵树k-multihedgehog：

1-multihedgehog是其中一个点度数大于等于3，其它点度数均为1.

k-multihedgehog是在k-1multihedgehog的基础上，把所有度为1的点替换成一个1-multihedgehog并与原图相连。

---

题解：

就按照题意反过来模拟一遍就好了，把最外层的度为1的点一层层的删去。

我写的这份代码比较丑，细节和特判的情况会比较麻烦。

---

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+7;
struct Edge{
    int v,nxt;
    Edge(int v=0,int nxt=0):v(v),nxt(nxt){}
}e[N*2];
int p[N],edn;
void add(int u,int v){
    e[++edn]=Edge(v,p[u]);p[u]=edn;
    e[++edn]=Edge(u,p[v]);p[v]=edn;
}
int d[N],vis[N];
int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    int u,v;
    memset(p,-1,sizeof(p));edn=-1;
    memset(vis,-1,sizeof(vis));
    queue<int>q[2];
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);
        d[u]++,d[v]++;
    }
    if(k>=n){printf("No\n");return 0;}
    for(int i=1;i<=n;i++){
        if(d[i]==1){
            q[0].push(i);
            vis[i]=0;
        }
    }
    int ans=0;
    for(int t=0;;t^=1){
        bool flag=true;
        while(!q[t].empty()){
            int x=q[t].front();q[t].pop();
            if(d[x]==0) continue;
            int u;
            for(int i=p[x];~i;i=e[i].nxt){
                if(d[e[i].v]==0) continue;
                u=e[i].v;break;
            }
            if(vis[u]==t||d[u]<3){flag=false;break;}
            int cnt=0;
            for(int i=p[u];~i;i=e[i].nxt){
                int v=e[i].v;
                if(d[v]==1&&vis[v]==t) d[v]--,d[u]--,cnt++;
            }
            if(cnt<3){flag=false;break;}
            q[t^1].push(u);vis[u]=t^1;
        }
        if(!flag){ans=-1;break;}
        if(q[t^1].empty()) break;
        ans++;
    }
    if(ans==k) printf("Yes\n");
    else printf("No\n");
}