bzoj 2054: 疯狂的馒头

最新推荐文章于 2019-05-29 17:23:00 发布

原创最新推荐文章于 2019-05-29 17:23:00 发布 · 382 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#并查集

收录于

本文介绍了一种使用并查集进行优化查找的算法实现，通过递归压缩路径来提高查找效率。具体实现中，利用了数学公式动态生成节点间的连接，并通过按秩合并策略减少查找次数。

并查集加速查找

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=(n);i++)
using namespace std;
const int maxn=1000015;
int col[maxn],n,m,p,q,f[maxn];
inline int find(int x){return f[x]==x ? x : f[x]=find(f[x]);}
int main(){
    scanf("%d%d%d%d",&n,&m,&p,&q);
    rep(i,1,n+1)f[i]=i;
    for(int i=m;i>=1;i--){
        int l=(1ll*i*p+q)%n+1,r=(1ll*i*q+p)%n+1;
        if(l>r)swap(l,r);
        for(int j=find(l);j<=r;j=find(f[j])){
            col[j]=i;
            f[j]=j+1;    ///
        }
    }
    rep(i,1,n)printf("%d\n",col[i]);
    return 0;
}