【BZOJ4320】Homework-分块+并查集

测试地址：Homework
做法：本题需要用到分块+并查集。
看到取模，一般的数据结构肯定做不了，于是考虑对权值序列分块。
首先，当模数不超过$\sqrt n$（$n$为最大数）时，我们显然可以开$\sqrt n$个桶，每次插入暴力更新这些桶，这样答案就可以每次$O(1)$地求出了。
然后，当模数超过$\sqrt n$时，它在$n$之内的倍数不超过$\sqrt n$个，那么取模后最小的数，也就是这些倍数点右边最近的点。怎么求这样的点呢？正向考虑比较麻烦，注意到题目可以离线，于是反向把加数变成减数，这样每删除一个数就意味着合并两个区间，就可以用并查集维护了。于是我们就解决了这一题，时间复杂度为$O(n\sqrt n)$。
以下是本人代码：

#include <bits/stdc++.h>
using namespace std;
int n,blocksiz,q,op[100010],opx[100010],fa[300010];
int ans[100010],nowans[610];
bool vis[300010]={0};

int find(int x)
{
    int r=x,i=x,j;
    while(r!=fa[r]) r=fa[r];
    while(i!=r) j=fa[i],fa[i]=r,i=j;
    return r;
}

void merge(int x,int y)
{
    int fx=find(x),fy=find(y);
    if (fx>fy) swap(x,y);
    fa[fx]=fy;
}

int main()
{
    n=300000;
    blocksiz=(int)sqrt(n)+1;
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {
        char s[5];
        scanf("%s",s);
        if (s[0]=='A') op[i]=0;
        else op[i]=1;
        scanf("%d",&opx[i]);
    }

    for(int i=1;i<=blocksiz;i++)
        nowans[i]=n+1;
    for(int i=1;i<=q;i++)
    {
        if (!op[i])
        {
            vis[opx[i]]=1;
            for(int j=1;j<=blocksiz;j++)
                nowans[j]=min(nowans[j],opx[i]%j);
        }
        else if (opx[i]<=blocksiz)
            ans[i]=nowans[opx[i]];
    }
    int last=n+1;
    fa[n+1]=n+1;
    for(int i=n;i>=0;i--)
    {
        if (vis[i]) last=i;
        fa[i]=last;
    }
    for(int i=q;i>=1;i--)
    {
        if (!op[i]) merge(opx[i],opx[i]+1);
        else if (opx[i]>blocksiz)
        {
            ans[i]=n+1;
            for(int j=0;j<=n;j+=opx[i])
                if (find(j)!=n+1)
                    ans[i]=min(ans[i],find(j)%opx[i]);
        }
    }
    for(int i=1;i<=q;i++)
        if (op[i]) printf("%d\n",ans[i]);

    return 0;
}