Codevs 2054疯狂的馒头

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;
}