P3942

Posted 2022-10-15 vimin

题目传送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;