2054:疯狂的馒头/2375:疯狂的染色

题目链接

题目大意：给定染色的顺序，每次染一个区间,每次染色会覆盖之前颜色,输出m次涂色后的每个点的颜色

题解：只有最后一次有用，所以倒着做
用f[i]=x来表示i到x-1这段区间已经被染过色了。当染过色之后，f[i]=i+1。这样就可以跳过已染色的区间，一直顺着f[i]染了。

我的收获：第一道双倍经验题……感觉用并查集维护区间的姿势还是挺不错的

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#define M 10000000+5
using namespace std;
#define ll long long
int n,m,p,q,f[M],c[M];

int fid(int x){return f[x]==x?x:f[x]=fid(f[x]);}

void work()
{
    int tot=0;//已染色节点数
    for(int i=m;i>=1;i--)
    {
        int l=((ll)i*p+q)%n+1,r=((ll)i*q+p)%n+1;//用long long防止中间运算溢出 
        if (l>r) swap(l,r);//不解释…… 
        for(int j=fid(l);j<=r;j=fid(j))//fid（l）就是f[l]，表示l及l后还没有染色的区间开始节点
        {
            c[j]=i;//给j染色
            tot++;
            f[j]=j+1;//j染色完毕，f[j]指向下一个未染色节点j+1
        }
        if (tot==n) break;//已经完成染色，退出循环 
    }
    for(int i=1;i<=n;i++)
        printf("%d\n",c[i]);
}

void init()
{
    scanf("%d%d%d%d",&n,&m,&p,&q);
    for(int i=1;i<=n+1;i++) f[i]=i;//f[n]可以指向n+1，所以要到n+1
}

int main()
{
    init();
    work();
    return 0;
}