【bzoj2054】 疯狂的馒头 并查集

一种并查集的应用，快速寻找下一个位置，当时考试的时候自己YY出来了。

#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<iostream>#include<algorithm>#define maxn 1000100using namespace std;int f[maxn],a[maxn];int n,m,p,q;int find(int x){if (f[x]==x) return x;else return f[x]=find(f[x]);}int main(){scanf("%d%d%d%d",&n,&m,&p,&q);for (int i=1;i<=n+1;i++) f[i]=i;for (int i=m;i>=1;i--){int x=((long long)i*p+q)%n+1,y=((long long)i*q+p)%n+1;if (x>y) swap(x,y);for (int j=find(x);j<=y;j=find(j)) a[j]=i,f[j]=j+1;}for (int i=1;i<=n;i++) printf("%d\n",a[i]);return 0;}