BZOJ4028 HEOI2015公约数数列（分块）

　　前缀gcd的变化次数是log的，考虑对每一种gcd查询，问题变为查询一段区间是否存在异或前缀和=x/gcd。

　　无修改的话显然可以可持久化trie，但这玩意实在没法支持修改。于是考虑分块。

　　对于每一块将其中所有块内异或前缀和排序。查询时先看这块与上一块相比gcd有没有变化，如果有对其中每个位置暴力查询，否则在排序后的数组中二分。修改时暴力改每一块的前缀gcd及异或和，被修改的块暴力重构排序数组即可。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
#define ll long long
ll read()
{
    ll x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
int n,m,a[N];
int block_size,block_tot,L[N],R[N],pos[N];
int gcd_pre[N],xor_pre[N],gcd_block[N],xor_block[N];
struct data
{
    int x,i;
    bool operator <(const data&a) const
    {
        return x<a.x||x==a.x&&i<a.i;
    }
}v[N];
int gcd(int n,int m){return m==?n:gcd(m,n%m);}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj4028.in","r",stdin);
    freopen("bzoj4028.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    n=read();
    for (int i=;i<=n;i++) a[i]=read();
    block_size=sqrt(n);block_tot=(n-)/block_size+;
    for (int i=;i<=block_tot;i++)
    L[i]=(i-)*block_size+,R[i]=min(n,i*block_size);
    for (int i=;i<=block_tot;i++)
    {
        for (int j=L[i];j<=R[i];j++)
        gcd_block[i]=gcd(gcd_block[i],a[j]),
        xor_block[i]=xor_block[i]^a[j],
        pos[j]=i;
        gcd_pre[i]=gcd(gcd_pre[i-],gcd_block[i]),xor_pre[i]=xor_pre[i-]^xor_block[i];
    }
    for (int i=;i<=block_tot;i++)
    {
        v[L[i]].x=a[L[i]],v[L[i]].i=L[i];
        for (int j=L[i]+;j<=R[i];j++)
        v[j].x=v[j-].x^a[j],v[j].i=j;
        sort(v+L[i],v+R[i]+);
    }
    m=read();
    while (m--)
    {
        char c=getchar();
        while (c<'A'||c>'Z') c=getchar();
        if (c=='M')
        {
            int p=read()+,x=read();a[p]=x;p=pos[p];
            gcd_block[p]=xor_block[p]=;
            for (int i=L[p];i<=R[p];i++)
            gcd_block[p]=gcd(gcd_block[p],a[i]),
            xor_block[p]=xor_block[p]^a[i];
            for (int i=p;i<=block_tot;i++)
            gcd_pre[i]=gcd(gcd_pre[i-],gcd_block[i]),
            xor_pre[i]=xor_pre[i-]^xor_block[i];
            v[L[p]].x=a[L[p]],v[L[p]].i=L[p];
            for (int i=L[p]+;i<=R[p];i++)
            v[i].x=v[i-].x^a[i],v[i].i=i;
            sort(v+L[p],v+R[p]+);
        }
        else
        {
            ll x=read();int ans=;
            for (int i=;i<=block_tot;i++)
            {
                if (ans) break;
                if (gcd_pre[i]==gcd_pre[i-])
                {
                    if (x%gcd_pre[i-]==)
                    {
                        int l=L[i],r=R[i];
                        while (l<=r)
                        {
                            int mid=l+r>>;
                            if (v[mid].x>=(x/gcd_pre[i-]^xor_pre[i-])) ans=mid,r=mid-;
                            else l=mid+;
                        }
                        if (ans&&v[ans].x==(x/gcd_pre[i-]^xor_pre[i-])) {ans=v[ans].i;break;}
                        else ans=;
                    }
                }
                else
                {
                    int p=gcd_pre[i-],q=xor_pre[i-];
                    for (int j=L[i];j<=R[i];j++)
                    {
                        p=gcd(p,a[j]),q^=a[j];
                        if (x%p==&&x/p==q) {ans=j;break;}
                    }
                }
            }
            if (ans) printf("%d\n",ans-);
            else printf("no\n");
        }
    }
    return ;
}