【bzoj2054】疯狂的馒头

Description

Input

第一行四个正整数N，M，p，q

Output

一共输出N行，第i行表示第i个馒头的最终颜色（如果最终颜色是白色就输出0）。

Sample Input

4 3 2 4
Sample Output

2

2

3

0
HINT

题解
倒着做，用并查集维护已经被覆盖的区间。

代码

#include<bits/stdc++.h>typedef long long ll;const int mod=1000000007;const int N=500005;using namespace std;inline int read(){    int x=0,f=1;char ch=getchar();    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int n,m,p,q,f[1000005],ans[1000005];int find(int i){    if (f[i]!=i) f[i]=find(f[i]);    return f[i];}int main(){    n=read();m=read();p=read();q=read();    for (int i=1;i<=n+1;i++) f[i]=i;    for (int i=m;i;i--)    {        int x=((ll)i*p+q)%n+1;        int y=((ll)i*q+p)%n+1;        if (x>y) swap(x,y);        for (int j=find(x);j<=y;j=find(j)){ans[j]=i;f[j]=j+1;}    }    for (int i=1;i<=n;i++) printf("%d\n",ans[i]);    return 0;}