关闭

【bzoj3123】 [Sdoi2013]森林

444人阅读 评论(0) 收藏 举报
分类:

  被教做人啦T_T
  维护树上的主席树。
  查询u,v,那么对应的当前版本是u,v,前驱版本是lca(u,v),fa[lca(u,v)]
  连u,v,因为保证是一棵树,于是启发式暴力合并,每次重新建主席树。
  时间复杂度O(Tlog2n)
  几个要注意的地方。
  第一个读入的testcases是逗你玩的尼玛
  倍增数组范围小的要放后面,唔这是常识然而为了代码少写几B居然就把它放前面去了。。。T了一晚上真是爽。。。
  

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a,_=b;i<=_;i++)
#define per(i,a,b) for(int i=a,_=b;i>=_;i--)
#define fore(i,u)  for(int i=head[u];i;i=nxt[i])
#define maxn 80007
#define maxm 170007
#define maxs 15000007

inline int rd()
{
    char c = getchar();
    while (!isdigit(c)) c = getchar() ; int x = c - '0';
    while (isdigit(c = getchar())) x = x * 10 + c - '0';
    return x;
}

typedef int arr[maxn];
typedef int adj[maxm];
typedef int seg[maxs];

adj to , nxt;
arr head , pa , sz , dep , rt , a , h;
seg lc , rc , cnt;

int fa[maxn][19];

int n , m , ett , tot , ans , all , T , mxd;

inline void ins(int u , int v)
{
    to[++ ett] = v , nxt[ett] = head[u] , head[u] = ett;
}

void update(int pr , int&nr , int l , int r , int v)
{
    if (!nr) nr = ++ tot;
    cnt[nr] = cnt[pr] + 1;
    if (l == r) return;
    int m = (l + r) >> 1;
    if (v <= m)
        rc[nr] = rc[pr] , update(lc[pr] , lc[nr] , l , m , v);
    else
        lc[nr] = lc[pr] , update(rc[pr] , rc[nr] , m + 1 , r , v);
}

void dfs1(int u)
{
    dep[u] ++;
    rep (i , 1 , mxd) fa[u][i] = fa[fa[u][i - 1]][i - 1];
    update(rt[fa[u][0]] , rt[u] , 1 , all , a[u]);
    fore (i , u)
    {
        int v = to[i];
        if (v == fa[u][0]) continue;
        dep[v] = dep[u] , fa[v][0] = u;
        dfs1(v);
    }
}

void dfs2(int u , int p)
{
    dep[u] = dep[p] + 1;
    fa[u][0] = p;
    rep (i , 1 , mxd) fa[u][i] = fa[fa[u][i - 1]][i - 1];
    update(rt[p] , rt[u] , 1 , all , a[u]);
    fore (i , u)
    {
        int v = to[i];
        if (v == p) continue;
        dfs2(v , u);
    }
}

int find(int u) { return u == pa[u] ? u : pa[u] = find(pa[u]) ; }

inline void _union(int u , int v)
{
    u = find(u) , v = find(v);
    if (u == v) return;
    if (sz[u] < sz[v]) swap(u , v);
    pa[v] = u , sz[u] += sz[v];
}

void input()
{
    n = rd();
    n = rd() , m = rd() , T = rd();
    mxd = (int) (log(n) / log(2)) + 1;
    rep (i , 1 , n) h[i] = a[i] = rd();
    sort(h + 1 , h + n + 1);
    all = unique(h + 1 , h + n + 1) - h - 1;
    rep (i , 1 , n) a[i] = lower_bound(h + 1 , h + all + 1 , a[i]) - h;
    rep (i , 1 , n) pa[i] = i , sz[i] = 1;
    rep (i , 1 , m)
    {
        int u = rd() , v = rd();
        ins(u , v) , ins(v , u);
        _union(u , v);
    }
    rep (i , 1 , n) if (!dep[i]) dfs1(i);
}

inline int lca(int u , int v)
{
    if (dep[u] < dep[v]) swap(u , v);
    int d = dep[u] - dep[v];
    rep (i , 0 , mxd) if ((d >> i) & 1) u = fa[u][i];
    if (u == v) return u;
    per (i , mxd , 0) if (fa[u][i] != fa[v][i]) u = fa[u][i] , v = fa[v][i];
    return fa[u][0];
}

inline void merge(int u , int v)
{
    ins(u , v) , ins(v , u);
    if (sz[u] < sz[v]) swap(u , v);
    _union(u , v);
    dfs2(v , u);
}

inline void query(int u , int v , int k)
{
    int x = lca(u , v) , p = fa[x][0];
    u = rt[u] , v = rt[v] , x = rt[x] , p = rt[p];
    int l = 1 , r = all;
    while (l < r)
    {
        int m = (l + r) >> 1;
        int w = cnt[lc[u]] + cnt[lc[v]] - cnt[lc[x]] - cnt[lc[p]];
        if (w >= k)
            u = lc[u] , v = lc[v] , x = lc[x] , p = lc[p] , r = m;
        else
            k -= w , u = rc[u] , v = rc[v] , x = rc[x] , p = rc[p] , l = m + 1;
    }
    printf("%d\n" , ans = h[l]);
}

void solve()
{
    char c;
    rep (i , 1 , T)
    {
        for (c = getchar();!isalpha(c);c = getchar()) ;
        int u = rd() ^ ans , v = rd() ^ ans;
        if (c == 'Q')
            query(u , v , rd() ^ ans);
        else
            merge(u , v);
    }
}

int main()
{
    input();
    solve();
    return 0;
}
0
0

猜你在找
【直播】机器学习&数据挖掘7周实训--韦玮
【套餐】系统集成项目管理工程师顺利通关--徐朋
【直播】3小时掌握Docker最佳实战-徐西宁
【套餐】机器学习系列套餐(算法+实战)--唐宇迪
【直播】计算机视觉原理及实战--屈教授
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之矩阵--黄博士
【套餐】微信订阅号+服务号Java版 v2.0--翟东平
【直播】机器学习之凸优化--马博士
【套餐】Javascript 设计模式实战--曾亮
查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:76783次
    • 积分:2169
    • 等级:
    • 排名:第17409名
    • 原创:136篇
    • 转载:0篇
    • 译文:0篇
    • 评论:44条
    小伙伴
    大神犇
    最新评论