树链剖分求LCA的模板(洛谷P3379)

2020.5.6
题目描述我就不写了,模板题。

之前写过倍增+tarjan求lca的板子,但是倍增这东西我貌似还没太搞懂,只写过st表,可能是我太菜了呜呜呜。

树剖其实我之前也没太弄懂,后来做了一道河南某年的省选题就豁然开朗了(但还没ac,可能是我用的线段树板子出问题了)。就是用2遍dfs处理好树内边的归属关系 + 不断向上溯祖,追溯到同一个祖先即是答案。top数组存放子树顶节点编号,depth存放节点所处的层数,fa数组是存节点的上一个节点,son是重儿子,言外之意就是子树中size最大的一个子书根节点,然后size就是用来存子树底下的节点数量。

这是lca,不过感觉树剖和线段树经常是配合使用的,因为一般都是要区间更新,所以一个一个查O(n)太慢了,需要先dfs序重新编号,然后建立线段树,用一个id数组记录编号,线段树就能直接更新了,感觉还是很神奇的算法。

代码如下,前向星记得数组开大,这题真变态,我开了40w还是不够,后来100w就ac了,毒瘤。

#include <bits/stdc++.h>
using namespace std;
#define limit (1000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%lld\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a ; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\administrator01\\CLionProjects\\untitled24\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}
int n, k,cnt;
int head[limit],depth[limit],fa[limit],top[limit],sizes[limit],son[limit];
struct node{
    int to, next;
}edge[limit<<2];
void init(){
    cnt = 0;
    memset(head, -1, sizeof(head));
}
void add(int u, int v){
    edge[cnt].to = v;
    edge[cnt].next = head[u];
    head[u] = cnt++;
}
void dfs(int u, int pre){
    fa[u] = pre;
    depth[u] = depth[pre] + 1;
    sizes[u] = 1;
    for(int i = head[u]; ~i; i = edge[i].next){
        int v = edge[i].to;
        if(v != pre){
            dfs(v,u);
            sizes[u] += sizes[v];
            if(sizes[son[u]] < sizes[v])son[u] = v;//剖分
        }
    }
}
void dfs2(int u, int topp){
    top[u] = topp;
    if(son[u]){
        dfs2(son[u],topp);
    }
    for(int i = head[u]; ~i; i = edge[i].next){
        int v = edge[i].to;
        if(v != fa[u] && v != son[u]){
            dfs2(v,v);
        }
    }

}
int query(int x,int y){
    while(top[x] != top[y]){
        if(depth[top[x]] >= depth[top[y]])x = fa[top[x]];
        else y = fa[top[y]];
    }
    return depth[x] < depth[y] ? x : y;
}
int main(){
#ifdef LOCAL
    FOPEN;
    //freopen("C:\\Users\\administrator01\\CLionProjects\\untitled24\\out.txt", "w", stdout);
#endif
    n = read(), k =read();
    int root = read();
    init();
    rep(i ,1, n-1){
        int x = read(), y = read();
        add(x,y);
        add(y,x);
    }
    dfs(root,0);
    dfs2(root, root);
    rep(i ,1, k){
        int x= read(), y = read();
        ff(query(x,y))
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章