Codevs 2054疯狂的馒头

最新推荐文章于 2017-10-13 22:03:00 发布

原创最新推荐文章于 2017-10-13 22:03:00 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#编程

题解同时被 3 个专栏收录

186 篇文章

订阅专栏

并查集

20 篇文章

订阅专栏

CodeVS

14 篇文章

订阅专栏

本文介绍了一个使用并查集数据结构解决馒头染色问题的算法。通过对馒头进行倒序染色，并利用并查集维护染色状态，有效地解决了问题。文中提供了一段C++实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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