P2391 白雪皑皑

原题链接

考察:并查集

并查集维护序列的连通性

Y总在课上提过的题目,本蒟蒻果然不会

思路:

       本质是快速染色某个区间.并查集可以帮助我们快速找到第一个未染色的点.

       设p[i]为以i为起点第一个需要还未涂色的点(包括i) .通过findf(i)可以帮助我们找到第一个未染色的点.我们在for循环染色[l,r]区间时,i可以直接跳到findf(i)处.这样压缩了染色路径.

       一个阅读理解的问题是染色l ~ r之间的点.如果l>r 是swap而不是continue

#include <iostream> 
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1000010;
int n,m,ps,q,p[N],color[N]; 
int get(int i,int l,int r)
{
    return (i*l+r)%n+1;
}
int findf(int x) 
{
    if(p[x]!=x) p[x] = findf(p[x]);
    return p[x];
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&ps,&q);
    for(int i=1;i<=n+5;i++) p[i] = i;
    for(int i=m;i>=1;i--)
    {
        int l = get(i,ps,q),r = get(i,q,ps);
        if(l>r) swap(l,r);
        l = findf(l);
        while(l<=r)
        {
            color[l] = i;
            p[l] = findf(l+1);
            l = findf(l+1);
        }
    }
    for(int i=1;i<=n;i++)
        printf("%d\n",color[i]);
    return 0;
}