zoukankan      html  css  js  c++  java
  • bzoj 3123 [Sdoi2013]森林(主席树+启发式合并+LCA)

          

    Description

    Input

    第一行包含一个正整数testcase,表示当前测试数据的测试点编号。保证1≤testcase≤20。 
    第二行包含三个整数N,M,T,分别表示节点数、初始边数、操作数。第三行包含N个非负整数表示 N个节点上的权值。 
     接下来 M行,每行包含两个整数x和 y,表示初始的时候,点x和点y 之间有一条无向边, 接下来 T行,每行描述一个操作,格式为“Q x y k”或者“L x y ”,其含义见题目描述部分。

    Output

    对于每一个第一类操作,输出一个非负整数表示答案。 
     
     

    Sample Input

    1
    8 4 8
    1 1 2 2 3 3 4 4
    4 7
    1 8
    2 4
    2 1
    Q 8 7 3 Q 3 5 1
    Q 10 0 0
    L 5 4
    L 3 2 L 0 7
    Q 9 2 5 Q 6 1 6

    Sample Output

    2
    2
    1
    4
    2

    HINT



    对于第一个操作 Q 8 7 3,此时 lastans=0,所以真实操作为Q 8^0 7^0 3^0,也即Q 8 7 3。点8到点7的路径上一共有5个点,其权值为4 1 1 2 4。这些权值中,第三小的为 2,输出 2,lastans变为2。对于第二个操作 Q 3 5 1 ,此时lastans=2,所以真实操作为Q 3^2 5^2 1^2 ,也即Q 1 7 3。点1到点7的路径上一共有4个点,其权值为 1 1 2 4 。这些权值中,第三小的为2,输出2,lastans变为 2。之后的操作类似。 

     

     
    思路:
    强制在线,第一个操作求树上第k小,主席树+LCA就好了,另一个操作连接两棵树,启发式合并就好了,这道题占的空间很大,主席树要开大点,之前开了100倍还是RE了。
     
    实现代码:
    #include<bits/stdc++.h>
    using namespace std;
    #define ll long long
    #define mid int m = (l + r) >> 1
    
    const int M = 1e5 + 10;
    struct node{
        int to,next;
    }e[M<<1];
    
    int idx,cnt,cnt1;
    int c[M],head[M],sum[M*200],root[M],ls[M*200],rs[M*200],p[M][30],dep[M],fa[M],v[M];
    int a[M],b[M],ans[M];
    void add(int u,int v){
        e[++cnt1].to = v;e[cnt1].next = head[u];head[u] = cnt1;
    }
    
    void pushup(int rt){
        sum[rt] = sum[ls[rt]] + sum[rs[rt]];
    }
    
    void update(int p,int l,int r,int old,int &rt){
        rt = ++idx; ls[rt] = ls[old]; rs[rt] = rs[old];
        sum[rt] = sum[old] + 1;
        if(l == r) return ;
        mid;
        if(p <= m) update(p,l,m,ls[old],ls[rt]);
        else update(p,m+1,r,rs[old],rs[rt]);
        pushup(rt);
    }
    
    int query(int old,int now,int lc,int flc,int l,int r,int k){
        if(l == r)  return l;
        mid;
        int ret = sum[ls[old]] + sum[ls[now]] - sum[ls[lc]] - sum[ls[flc]];
        if(ret >= k) query(ls[old],ls[now],ls[lc],ls[flc],l,m,k);
        else query(rs[old],rs[now],rs[lc],rs[flc],m+1,r,k-ret);
    }
    
    int Find(int x){
        if(x == fa[x]) return x;
        fa[x] = Find(fa[x]);
        return fa[x];
    }
    
    void dfs(int u,int faz){
        dep[u] = dep[faz] + 1;
        for(int i = 1;i < 20;i ++)
            p[u][i] = p[p[u][i-1]][i-1];
        update(v[u],1,cnt,root[faz],root[u]);
        for(int i = head[u];i;i=e[i].next){
            int v = e[i].to;
            if(v == faz) continue;
            p[v][0] = u;
            dfs(v,u);
        }
    }
    
    int lca(int a,int b){
        if(dep[a] > dep[b]) swap(a,b);
        int h = dep[b] - dep[a];
        for(int i = 0;(1<<i)<=h;i ++){
            if((1<<i)&h) b = p[b][i];
        }
        if(a!=b){
            for(int i = 22;i >= 0;i --){
                if(p[a][i] != p[b][i]){
                    a = p[a][i]; b = p[b][i];
                }
            }
            a = p[a][0];
        }
        return a;
    }
    
    int main()
    {
        int t,n,m;
        int last = 0;
        scanf("%d",&t);
        scanf("%d%d%d",&n,&m,&t);
        for(int i = 1;i <= n;i ++){
            scanf("%d",&a[i]);
            b[i] = a[i];
        }
        sort(b+1,b+1+n);
        cnt = unique(b+1,b+1+n)-b-1;
        for(int i = 1;i <= n;i ++){
            v[i] = lower_bound(b+1,b+1+cnt,a[i]) - b;
            ans[v[i]] = a[i];
        }
    
        for(int i = 1;i <= n;i ++){
            fa[i] = i; c[i] = 1;
        }
        int u,v,k;
        for(int i = 1;i <= m;i ++){
            scanf("%d%d",&u,&v);
            add(u,v); add(v,u);
            int fx = Find(u),fy = Find(v);
            if(fx != fy){
                fa[fy] = fx;
                c[fx] += c[fy];
            }
        }
        for(int i = 1;i <= n;i ++){
            if(!dep[i]) dfs(Find(i),0);
        }
        char op[10];
        for(int i = 1;i <= t;i ++){
            scanf("%s",op);
            if(op[0] == 'Q'){
                scanf("%d%d%d",&u,&v,&k);
                u ^= last; v ^= last; k^= last;
                int lc = lca(u,v);
                int num = query(root[u],root[v],root[lc],root[p[lc][0]],1,cnt,k);
                last = ans[num];
                printf("%d
    ",last);
            }
            else{
                scanf("%d%d",&u,&v);
                u ^= last; v ^= last;
                add(u,v); add(v,u);
                int fx = Find(u),fy = Find(v);
                if(c[fx] < c[fy]){
                    swap(u,v); swap(fx,fy);
                }
                fa[fy] = fx; c[fx] += c[fx];
                p[v][0] = u; dfs(v,u);
            }
        }
        return 0;
    }
  • 相关阅读:
    GitHub的Release下载文件中断解决方案
    Nacos笔记(一):下载安装
    【深度学习】torch中LSTM的参数详细情况
    item 安装 oh-my-zsh
    Azure IoT Edge入门(10)远程在IoT Edge设备上部署SQL数据库-Deploy Azure SQL Edge to Azure IoT Edge
    Azure IoT Edge入门(11)IoT Edge设备上的函数计算Function,使用Function写数据库-Using Azure Function On Edge device save data to Azure SQL Edge
    Azure IoT Edge入门(8)边缘设备到云的消息 及 在本地开发调试 debug Edge Module
    Azure IoT Edge入门(9)云到边缘设备控制-Azure iot edge module direct method
    通过Azure bot framework composer 设计一个AI对话机器人bot(查询天气)
    VSCode Hot Exit
  • 原文地址:https://www.cnblogs.com/kls123/p/10031976.html
Copyright © 2011-2022 走看看