关闭

[分块+并查集按秩合并]#519. 「LibreOJ β Round #2」数学上来先打表

标签: LOJ
629人阅读 评论(0) 收藏 举报
分类:

题目梗概

Angle Beats,番,仲村由理(小百合~)

给你一个图,每个点有点权,最开始没有边。

有一些操作:

  1. 添加一条 x 与 y 之间的双向边。
  2. 回到第 x 次操作后的状态。(注意这里的 x 可以是 0,即回到初始状态)
  3. 查询 x 所在联通块能到的点中点权第 y 小的值,如果不存在,那么输出 −1。

解题思路

对于第2个操作,将询问建成树后DFS。

对于第3个操作,对于每个联通块分块维护前缀和,对于当前块暴力查询一个点是否出现。

在建边时,需要按秩合并两个并查集。

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
inline int _read(){
    int num=0;char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num;
}
const int maxn=100005,S=1200;
int n,m,a[maxn],id[maxn],sum[maxn][maxn/S+3],c[maxn],s[maxn],f[maxn],p[maxn],X[maxn],Y[maxn],ans[maxn],K;
vector<int> d[maxn];
bool cmp(int x,int y){return a[x]<a[y];}
int get(int x){if (f[x]==x) return x;return get(f[x]);}
int query(int x,int y){
    if (y>s[x]) return 0;
    int t=0;while(y>sum[x][t]) y-=sum[x][t++];
    for (int i=0;i<S;i++){
        y-=(get(id[t*S+i])==x);
        if (!y) return id[S*t+i];
    }
}
void DFS(int x){
    if (p[x]==1){
        int fx=get(X[x]),fy=get(Y[x]);
        if (s[fx]>s[fy]) swap(fx,fy);
        if (fx!=fy){
            f[fx]=fy;s[fy]+=s[fx];
            for (int i=0;i<=K;i++) sum[fy][i]+=sum[fx][i];
        }
        for (int i=0;i<d[x].size();i++) DFS(d[x][i]);
        if (fx!=fy){
            f[fx]=fx;s[fy]-=s[fx];
            for (int i=0;i<=K;i++) sum[fy][i]-=sum[fx][i];
        }
    }else{
        if (p[x]==3) ans[x]=query(get(X[x]),Y[x]);
        for (int i=0;i<d[x].size();i++) DFS(d[x][i]);
    }
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();m=_read();a[0]=-1;K=n/S+1;
    for (int i=1;i<=n;i++) a[i]=_read(),id[i]=i;
    sort(id+1,id+1+n,cmp);
    for (int i=1;i<=n;i++) c[i]=a[id[i]],f[i]=i,s[i]=1,sum[id[i]][i/S]=1;
    for (int i=1;i<=m;i++){
        p[i]=_read();X[i]=_read();
        if (p[i]==2) d[X[i]].push_back(i);else{
            Y[i]=_read();
            d[i-1].push_back(i);
        }
    }
    DFS(0);
    for (int i=1;i<=m;i++) if (p[i]==3) printf("%d\n",a[ans[i]]);
    return 0;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:54758次
    • 积分:2110
    • 等级:
    • 排名:千里之外
    • 原创:149篇
    • 转载:1篇
    • 译文:0篇
    • 评论:45条
    我的日记
    博客专栏