题目传送P3942

思路

• 是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;
 } 

原文地址：https://www.cnblogs.com/Vimin/p/11508242.html