BZOJ2054疯狂的馒头

2054: 疯狂的馒头
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 634 Solved: 250
Description

Input
第一行四个正整数N，M，p，q
Output
一共输出N行，第i行表示第i个馒头的最终颜色（如果最终颜色是白色就输出0）。
Sample Input
4 3 2 4
Sample Output
2
2
3
0
HINT

首先考虑数据范围，区间修改的话用线段树对于10^7的数据肯定会T掉，所以需要一个优化的算法。。
由于求最后的染色，所以可以用并查集。。
怎么用呢。。
从后往前枚举，如果这个点已经被染过色了，那么下面的操作便不需要染色了。。
这样每个点就会只染色一次。。
就在并查集中将这个点连向右面那个。。
时间复杂度O（n+m）。。
附上本蒟蒻的代码：

#include<cstdio>
#include<iostream>
using namespace std;
int father[10000002],a[1000001];

int read()
{
    int w=0;
    char ch=getchar();
    while (ch>='0' && ch<='9')
      {
        w=w*10+ch-'0';
        ch=getchar();
      }
    return w;
}

int find(int f)//这里要写递归的路径压缩
{
    if (!father[f] || father[f]==f)
      return father[f]=f;
    return father[f]=find(father[f]);
}

int main()
{
    int n,m,p,q,i,j,x,y;
    n=read();
    m=read();
    p=read();
    q=read();
    for (i=m;i>=1;i--)
      {
        x=((long long)i*p+q)%n+1;
        y=((long long)i*q+p)%n+1;
        if (x>y)
          swap(x,y);
        for (j=find(x);j<=y;j=find(j))
          {
            a[j]=i;
            father[j]=j+1;
          }
      }
    for (i=1;i<=n;i++)
      printf("%d\n",a[i]);
    return 0;
}