bzoj2054 疯狂的馒头

题目

区间操作，首选线段树，但是这道题显然不可行。

怎么办，我们考虑如果从后往前的话，一个数最多被修改一次，并查集就派上用场了。

看代码

#include<bits/stdc++.h>
#define N 10000000
using namespace std;
int n,m,p,q;
int f[N+1],a[N+1];
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void write(int x)
{
    if(x==0)putchar('0');
    else
    {
        char buf[15];
        int len=0;
        if(x<0)putchar('-'),x=-x;
        while(x)buf[++len]=x%10+'0',x/=10;
        for(int i=len;i>=1;i--)putchar(buf[i]);
    }
    putchar('\n');
}
inline int find(int x)
{
    int tmp,i,rt;
    rt=x; 
    while(rt!=f[rt])
        rt=f[rt];
    i=x;
    while(i!=rt)
    {
        tmp=f[i];
        f[i]=rt;
        i=tmp;
    }
    return rt;
}
int main()
{
    freopen("in.txt","r",stdin);
    n=read(),m=read(),p=read(),q=read();
    for(int i=1;i<=n;i++)f[i]=i;
    for(int i=m;i>=1;i--)
    {
        int x=((long long)i*p+q)%n+1;
        int y=((long long)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;
    }
    for(int i=1;i<=n;i++)
        write(a[i]);
    return 0;
}

比较坑的是，递归写会爆栈233

还有long long 强转 int 慢bi了。