【BZOJ 2054】 2054: 疯狂的馒头 (并查集特技)

Input

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

Output

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

Sample Input

4 3 2 4

Sample Output

2
2
3
0

HINT

 

 

 

【分析】

　　想要不带log，就从后往前做，每个点只染色一次，用神奇的并查集维护即可。

 

 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 using namespace std;
 #define Maxn 1000010

 int col[Maxn],rt[Maxn];

 int rtt(int x)
 {
     if(rt[x]!=x) rt[x]=rtt(rt[x]);
     return rt[x];
 }

 int tot=;
 void solve(int l,int r,int nw)
 {
     while(l<=r)
     {
         if(col[l]==)
         {
             col[l]=nw;
             rt[l]=rtt(l+);
             l=rtt(l);
             tot++;
         }
         l=rtt(l);
     }
 }

 int main()
 {
     int n,m,p,q;
     scanf("%d%d%d%d",&n,&m,&p,&q);
     memset(col,,sizeof(col));
     for(int i=;i<=n+;i++) rt[i]=i;
     for(int i=m;i>=;i--)
     {
         if(tot==n) break;
         int l=(i*p+q)%n+,r=(i*q+p)%n+;
         if(l<=r) solve(l,r,i);
         else solve(r,l,i);
     }
     for(int i=;i<=n;i++) printf("%d\n",col[i]);
     return ;
 }

还有就是不要理解错那个区间。。[r,l]也是可以的。

2017-03-06 20:57:21