题目传送门sx
最后染色的馒头一定不会再被更改,所以倒着来染色;
用并查集来维护染了的区间,直接跳到祖先
因为染色是连续的点,把染了色的点的祖先设成右端点;
在区间内逐个染色,如果该点被染色,就把指针直接跳到该点的祖先;
如果跳到的祖先超出了r,别忘了跳出循环
注意左端点小于等右端点
#include <iostream>
#include <cstdio>
using namespace std;
const int N=1e6+2;
int n,m,fa[N],p,q,ans[N];
void co(int l,int r,int col)
{
for(int i=l;i<=r;i++){
while(ans[i]) i=fa[i]+1;
if(i>r) break;
fa[i]=fa[r];
ans[i]=col;
}
}
int main()
{
scanf("%d%d%d%d",&n,&m,&p,&q);
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=m;i;--i){
int x=(i*p+q)%n+1,y=(i*q+p)%n+1;
if(x>y) swap(x,y);
co(x,y,i);
}
for(int i=1;i<=n;i++)
printf("%d\n",ans[i]);
return 0;
}