bzoj2054 疯狂的馒头

2054: 疯狂的馒头

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 715   Solved: 298
[ Submit][ Status][ Discuss]

Description

Input

第一行四个正整数N，M，p，q

Output

一共输出N行，第i行表示第i个馒头的最终颜色（如果最终颜色是白色就输出0）。

Sample Input

4 3 2 4

Sample Output

2
2
3
0

HINT

并查集

将操作倒着处理，相当于每个馒头至多被染一次，每次只会染到空白的馒头。

用并查集维护每个馒头后面第一个白馒头(包括自己)，然后每次染色暴力染区间内所有空馒头，同时把染了色的馒头的集合和后面一个的集合进行合并。

和bzoj4551类似，用并查集维护离自己最近的满足某种性质的位置，然后每次修改后和后面一个合并，这种思路很好。

时间复杂度O(n+m)。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define N 1000005
#define M 10000005
using namespace std;
int n,m,p,q,a[N],f[N];
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
inline int find(int x)
{
	return f[x]==x?x:f[x]=find(f[x]);
}
int main()
{
	n=read();m=read();p=read();q=read();
	F(i,1,n+1) f[i]=i;
	D(i,m,1)
	{
		int x=(i*p+q)%n+1,y=(i*q+p)%n+1;
		if (x>y) swap(x,y);
		for(int j=find(x);j<=y;j=find(j)) a[j]=i,f[j]=j+1;
	}
	F(i,1,n) printf("%d\n",a[i]);
	return 0;
}