BZOJ 4281 [ONTAK2015]Związek Harcerstwa Bajtockiego 树上倍增

看这题吓得我赶紧写了一遍。。。
4s+还有RK 19。。

#include <cstdio>
#include <algorithm>
using namespace std;
#define FOR(i,j,k) for(int i=j;i<=k;++i)
const int N = 1000005, M = N * 2;
int h[N], p[M], v[M], dep[N], fa[N][21], K = 20, cnt = 0;
void add(int a, int b) { p[++cnt] = h[a]; v[cnt] = b; h[a] = cnt; }
void dfs(int x, int f) {
    fa[x][0] = f; dep[x] = dep[f] + 1;
    FOR(i,1,K) fa[x][i] = fa[fa[x][i - 1]][i - 1];
    for (int i = h[x]; i; i = p[i])
        if (v[i] != f) dfs(v[i], x);
}
int jump(int &x, int d) {
    FOR(i,0,K) if ((1 << i) & d) x = fa[x][i];
    return x;
}
int lca(int x, int y) {
    if (dep[x] < dep[y]) swap(x, y);
    jump(x, dep[x] - dep[y]);
    for(int i=K;i>=0;--i) if (fa[x][i] != fa[y][i]) x = fa[x][i], y = fa[y][i];
    return x == y ? x : fa[x][0];
}

int main() {
    int n, now, k, d, l, x, y;
    scanf("%d%d%d", &n, &now, &k);  
    FOR(i,2,n) scanf("%d%d", &x, &y), add(x, y), add(y, x);
    dfs(1, 0);  
    FOR(i,1,k) {
        scanf("%d%d", &d, &l);
        int f = lca(now, d);  
        if (dep[d] + dep[now] - 2 * dep[f] <= l) now = d;
        else {
            if (dep[now] - dep[f] >= l) jump(now, l);
            else {
                jump(d, dep[d] - dep[f] - (l - (dep[now] - dep[f])));
                now = d;
            }
        }
        printf("%d ", now);
    }
    return 0;
}

4281: [ONTAK2015]Związek Harcerstwa Bajtockiego

Description

给定一棵有n个点的无根树，相邻的点之间的距离为1，一开始你位于m点。之后你将依次收到k个指令，每个指令包含两个整数d和t，你需要沿着最短路在t步之内（包含t步）走到d点，如果不能走到，则停在最后到达的那个点。请在每个指令之后输出你所在的位置。

Input

第一行包含三个正整数n,m,k(1<=m<=n<=1000000,1<=k<=1000000)。
接下来n-1行，每行包含两个正整数x,y(1<=x,y<=n)，描述一条树边。
接下来k行，每行两个整数d,t(1<=d<=n,0<=t<=10^9)，描述一条指令。

Output

输出一行，包含k个正整数，即执行每条指令后你所在的位置。

Sample Input

3 1 2
1 2
2 3
3 4
1 1

Sample Output

3 2

HINT

Source

By Claris