每个区间只被覆盖一次,求每个点被哪种区间覆盖或者某个区间是否已经被覆盖过都可以用并查集做。
做法:每个点都指向当前被覆盖区间的右端点+1的位置,某个点的下一个没被覆盖的点是gf(i),同理如果某个区间[l,r]的gf(l)>=r+1,则这个区间已经被完全覆盖。
显然每个点只会被最后一次染色确定颜色,倒着做就好了。
#include<iostream> #include<cstring> #include<cstdlib> #include<cstdio> using namespace std; const int maxn=1000010; int n,m,p,q; int fa[maxn],col[maxn]; void read(int &k) { int f=1;k=0;char c=getchar(); while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar(); while(c<='9'&&c>='0')k=k*10+c-'0',c=getchar(); k*=f; } int gf(int x){return fa[x]==x?x:fa[x]=gf(fa[x]);} int main() { read(n);read(m);read(p);read(q); for(int i=1;i<=n+1;i++)fa[i]=i; for(int i=m;i;i--) { int l=(1ll*i*p+q)%n+1,r=(1ll*i*q+p)%n+1; if(l>r)swap(l,r); for(int j=gf(l);j<=r;j=gf(j)) col[j]=i,fa[j]=gf(j+1); } for(int i=1;i<=n;i++)printf("%d ",col[i]); }