CQF 十分喜欢吃馒头。
兴奋之下他一下子买了 N 个馒头请所有认识他的人吃。
但是 CQF 不喜欢白色,喜欢红色、黄色、绿色等鲜艳的颜色。
于是他把所有白色的馒头排成一列。
然后进行 M 次染色操作。
每个染色操作都是用一个神奇的刷子把连续的多个馒头染成特定的某种颜色。
一个馒头最终的颜色是最后一次染它的颜色。
如果一个馒头没有被染过色,那么它的颜色就是白色。
现在 CQF 已经定好了染色计划:在第 ii次染色操作中,把第(i×p+q)modN+1 个馒头和第 (i×q+p)modN+1 个馒头之间的馒头染成颜色 ii,其中 p,q 是特定的两个正整数。
他想立即知道最后每个馒头的颜色。
你能帮他吗?
输入格式
第一行四个正整数 N,M,p,q。
输出格式
一共输出 NN 行,第 i行表示第 i 个馒头的最终颜色(如果最终颜色是白色就输出 0。
数据范围
在 20% 的数据中,1≤N≤103,1≤M≤104
在 40% 的数据中,1≤N≤104,1≤M≤105
在 60% 的数据中,1≤N≤5×104,1≤M≤5×105
在 80% 的数据中,1≤N≤3×105,1≤M≤3×106
在 100% 的数据中,1≤N≤106,1≤M≤107
保证所有输入数据中 1≤M×p+q,M×q+p≤231−1。
输入样例:
4 3 2 4
输出样例:
2
2
3
0
分析:这个题用到了并查集, 一个馒头最终的颜色是最后一次染它的颜色,因此我们可以从后往前染色,这样第一次染色就是这个馒头最后的颜色。我们并查集的fa数组定义成从该点开始包括当前点向右,第一个未染色的点的坐标,当一个点被染色后,它的父节点指向右边第一个还未染色的点,那对每次染色找到一个区间内可以染色的点然后更新颜色以及fa数组就ok啦~~~
但是在这里我有一个疑问我的代码中我写成下面这样就会超时
cin>>n>>m>>p>>q;
for(int i=1;i<=n;i++) f[i]=i;
然而下面这样就不会
for(int i=1;i<=N;i++) f[i]=i;
cin>>n>>m>>p>>q;
很奇怪,不知道什么原因
最终的AC代码如下:
#include <iostream>
using namespace std;
const int N=1e6+10;
int f[N],ans[N];
int n,m,p,q;
int find(int x)
{
if(x!=f[x])
{
f[x]=find(f[x]);
}
return f[x];
}
int main()
{
for(int i=1;i<=N;i++) f[i]=i;
cin>>n>>m>>p>>q;
while(m)
{
int a=(m*p+q)%n+1,b=(m*q+p)%n+1;
int l=min(a,b),r=max(a,b);
int x=find(l);
while(x<=r)
{
ans[x]=m;
f[x]=find(x+1);
x=find(x);
}
m--;
}
for (int i = 1; i <= n; i++)
printf("%d\n", ans[i]);
return 0;
}