【题目描述】:
CQF十分喜欢吃馒头,兴奋之下他一下子买了N 个馒头请所有认识他的人吃。
但是CQF不喜欢白色,喜欢红色、黄色、绿色等鲜艳的颜色。于是他把所有白色的馒头排成一列。然后进行M 次染色操作。每个染色操作都是用一个神奇的刷子把连续的多个馒头染成特定的某种颜色。一个馒头最终的颜色是最后一次染它的颜色。如果一个馒头没有被染过色,那么它的颜色就是白色。
现在CQF已经定好了染色计划:在第i次染色操作中,把第(i × p + q )mod N + 1个馒头和第(i × q + p)mod N + 1个馒头之间的馒头染成颜色i,其中p, q是特定的两个正整数。他想立即知道最后每个馒头的颜色。你能帮他吗?
【输入描述】:
第一行四个正整数N ,M ,p,q。
【输出描述】:
一共输出N 行,第i行表示第i个馒头的最终颜色(如果最终颜色是白色就输出0)。
【样例输入】:
4 3 2 4
【样例输出】:
2
2
3
0
【时间限制、数据范围及描述】:
时间:1s 空间:256M
在20%的数据中,1<=N<=1000,1<=M<=10000
在40%的数据中,1<=N<=10000,1<=M<=100000
在60%的数据中,1<=N<=50000,1<=M<=500000
在80%的数据中,1<=N<=300000,1<=M<=3000000
在100%的数据中,1<=N<=1000000,1<=M<=10000000
保证所以输入数据中1<=Mp + q、Mq + p<= 2^31-1
【解题思路】:
并查集水题。。。从m到1倒序染色,染色的同时将该点与右边合并,保证每个点只染色一次。
【AC代码】:
#include<bits/stdc++.h>
#define M(a,b) memset(a,b,sizeof(a))
using namespace std;
int fa[1000005],ans[1000005];
long long n,m,p,q,l,r,i,j;
inline void read(long long &x){
char ch=getchar(),c=ch;
x=0;
while(ch<'0' || ch>'9'){
c=ch;
ch=getchar();
}
while(ch>='0' && ch<='9'){
x=(x<<1)+(x<<3)+ch-'0';
ch=getchar();
}
if(c=='-')x=-x;
}
int find(int x){
int r=x,pre;
while(r!=fa[r])r=fa[r];
while(x!=r){
pre=fa[x];
fa[x]=r;
x=pre;
}
return r;
}
int main(){
read(n),read(m),read(p),read(q);
for(i=1;i<=n+2;i++)fa[i]=i;
for(i=m;i>=1;i--){
l=(i*p+q)%n+1;
r=(i*q+p)%n+1;
if(l>r)swap(l,r);
for(j=find(l);j<=r;j=find(j)){
ans[j]=i;
fa[j]=j+1;
}
}
for(i=1;i<=n;i++)printf("%d\n",ans[i]);
return 0;
}