Codeforces Round #199 (Div. 2) E. Xenia and Tree

题目链接

2了，差点就A了。。。这题真心不难，开始想的就是暴力spfa就可以，直接来了一次询问，就来一次的那种，TLE了，想了想，存到栈里会更快，交又TLE了。。无奈C又被cha了，我忙着看C去了。。。这题，是我一个地方写错了。。top = 0的时候会死循环吗？貌似不会把，反正我加了这个判断，就A了，可能优化了一下把。

#include <cstring>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <queue>
using namespace std;
#define INF 1000000
int first[200001];
int dis[200001];
int in[200001];
int s[200001];
int n,t,top;
struct node
{
    int u,v,next;
}edge[300001];
void CL()
{
    t = 0;
    memset(first,-1,sizeof(first));
}
void add(int u,int v)
{
    edge[t].u = u;
    edge[t].v = v;
    edge[t].next = first[u];
    first[u] = t ++;
}
void spfa()
{
    int i,u,v;
    for(i = 1;i <= n;i ++)
    in[i] = 0;
    queue<int>que;
    for(i = 0;i < top;i ++)
    {
        in[s[i]] = 1;
        dis[s[i]] = 0;
        que.push(s[i]);
    }
    while(!que.empty())
    {
        u = que.front();
        que.pop();
        in[u] = 0;
        for(i = first[u];i != -1;i = edge[i].next)
        {
            v = edge[i].v;
            if(dis[v] > dis[u] + 1)
            {
                dis[v] = dis[u] + 1;
                if(!in[v])
                {
                    in[v] = 1;
                    que.push(v);
                }
            }
        }
    }
    return ;
}
int main()
{
    int m,i,u,v;
    CL();
    scanf("%d%d",&n,&m);
    for(i = 1;i < n;i ++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    for(i = 1;i <= n;i ++)
    dis[i] = INF;
    top = 1;
    s[0] = 1;
    spfa();
    top = 0;
    for(i = 1;i <= m;i ++)
    {
        scanf("%d%d",&u,&v);
        if(u == 1)
        {
            s[top ++] = v;
        }
        else
        {
            if(top != 0)//这里
            spfa();
            top = 0;
            printf("%d
",dis[v]);
        }
    }
    return 0;
}