思路
- 是P2279消防局的设立的加强版
- 我们只需要每次找到最深的点，然后找到向上距离他 (k) 个深度的祖先
- 话说只要在之前的代码改几个地方就行
- 可我偏偏调了一晚上
- 数据到了1e5
- 到大数据就莫名死循环
- 改了下
- 还是TLE
- TLE的原因 是每次把选中点距离为 (k) 的点标记，要记录前一个点，而不能直接用它的父亲，因为并不准确，我们可能向下走也可能向上走，上一个走过来的不一定是父亲。需要单独记录。
- 死循化 的原因好像是弹出队列时死循环了
代码
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
#pragma GCC optimize(2)
using namespace std;
int n,tot,ans,k,t;
bool vis[100005];
int head[100005];
int dep[100005],fa[100005];
struct node{
int to,net;
}e[200005];
struct cmp{
bool operator () (int &a,int &b){
return dep[a]<dep[b];
}
};
priority_queue<int,vector<int>,cmp>q;
inline void add(int x,int y){
e[++tot].to =y,e[tot].net=head[x],head[x]=tot;
e[++tot].to =x,e[tot].net=head[y],head[y]=tot;
}
inline void dfs1(int x,int f,int depth){
fa[x]=f;
dep[x]=depth;
for(int i=head[x];i;i=e[i].net ){
int to=e[i].to ;
if(to==f) continue;
dfs1(to,x,depth+1);
}
}
inline void dfs2(int x,int from,int depth){
if(depth>k) return ;
vis[x]=1;
for(int i=head[x];i;i=e[i].net )
if(e[i].to !=from)dfs2(e[i].to ,x,depth+1);
}
void cal(int x){
int num=0;
while(num<k){
num++;
x=fa[x];
}
t=x;
/*if(x==0) {
t=x;
return ;
}
if(num==k){
t=x;
return ;
}
cal(fa[x],num+1);*/
}
inline int read(){
int s=0,w=1;char ch=getchar();
while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
int main()
{
// freopen("63.in","r",stdin);
// freopen("63.out","w",stdout);
n=read();k=read();t=read();
// scanf("%d%d%d",&n,&k,&t);
for(int i=1;i<n;i++){
int x,y;
x=read();y=read();
// scanf("%d%d",&x,&y);
add(x,y);
}
dfs1(1,0,1);
for(int i=1;i<=n;i++) q.push(i);
int x;
while(q.size()){
x=q.top() ;q.pop();
if(vis[x]) continue ;
// while(q.size()&&vis[x=q.top()]) q.pop();
// cout<<"DE"<<endl;
cal(x);
if(t) dfs2(t,t,0);
else dfs2(1,1,0);
ans++;
}
printf("%d",ans);
return 0;
}