存图,还是set好。。删除贼快(作死拿vector存了下试试,然后果断超时)
跑dfs也行
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+7;
int n,k,u,v,flag,du[maxn];
set<int> head[maxn];
map<int,int> M;
void add(){
head[u].insert(v);
head[v].insert(u);
}
int sovle(){
for(int i=1;i<=n;i++)
if(head[i].size() == 1)
M[i] = 1;
while(n>1){
map<int,int> Tm;
for(auto i:M){
int x = *head[i.first].begin();
Tm[x]++;
n--;
head[x].erase(i.first);
}
for(auto i:Tm)
if(i.second<3)
return 0*printf("No\n");
M = Tm;
k--;
}
return 0*printf("%s\n",k?"No":"Yes");
}
void input(){
cin>>n>>k;
for(int i=0;i<n-1;i++)
cin>>u>>v,add();
}
int main(){
input();
sovle();
return 0;
}