AC通道: http://www.lydsy.com/JudgeOnline/problem.php?id=2054
[分析]
对于任何一个馒头,只有最后一次刷才有意义。无论之前刷过多少次,它的颜色都是由最后一次刷决定。所以,我们可以考虑倒过来刷,如果已经刷过就不需要再刷了。我们可以用并查集维护这个链表,用fa[i]表示i右边第一个未刷过的的馒头。
核心代码:for(int j=find(l);j<=r;j=find(j)){color[j]=i;fa[j]=j+1;}
#include <iostream>
#include <cstdio>
using namespace std;
int n,m,p,q;
int fa[1000010];
int color[1000010];
int find(int x){
int tmp=x,pre;
while(tmp!=fa[tmp])tmp=fa[tmp];
while(x!=tmp){
pre=fa[x];
fa[x]=tmp;
x=pre;
}
return tmp;
}
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>=1;i--){
int l=(i*p+q)%n+1;
int r=(i*q+p)%n+1;
if(l>r){int t=l;l=r;r=t;}
for(int j=find(l);j<=r;j=find(j)){
if(j==0)break; //注意:如果不加这一条会导致超时
color[j]=i;fa[j]=j+1;
}
}
for(int i=1;i<=n;i++)printf("%d\n",color[i]);
return 0;
}