Description
Input
第一行四个正整数N,M,p,q
Output
一共输出N行,第i行表示第i个馒头的最终颜色(如果最终颜色是白色就输出0)。
Sample Input
4 3 2 4
Sample Output
2
2
3
0
2
3
0
HINT
题解:可以发现对于每个馒头只有最后一次修改有用。
所以我们倒着做,然后用并查集维护一下颜色块即可。
具体维护的时候把每个点向它右边那个点连边。
代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 1001001
using namespace std;
int n,m,p,q;
int a[M];
namespace Union_Find_Set{
int fa[M];
int Find(int x)
{
if(!fa[x]||fa[x]==x)
return fa[x]=x;
return fa[x]=Find(fa[x]);
}
}
int main()
{
using namespace Union_Find_Set;
int i,j;
cin>>n>>m>>p>>q;
for(i=m;i;i--)
{
int x=((long long)i*p+q)%n+1;
int 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,fa[j]=j+1;
}
for(i=1;i<=n;i++)
printf("%d\n",a[i]);
return 0;
}