bzoj 4028: [HEOI2015]公约数数列 （分块乱搞+map）

题目描述

传送门

题目大意:

题解

将数列分块，对于每个块维护块内的gcd前缀和xor前缀和，并用map维护出每个块内某一权值出现的位置。
所有的修改操作只会影响一个块，时间是$\sqrt n \log n$
对于查询操作，考虑到gcd最多变化log次，我们从前向后扫，lastgcd,lastxor表示到上一个块为止的gcd,xor前缀和。如果当前区间的gcd前缀和是lastgcd的倍数，那么说明这段区间的gcd都是lastgcd,直接对于块内进行查询,查询是否存在$(x/lastgcd)$^$lastxor$.
如果不是倍数，说明在当前块的区间中gcd发生了改变，直接暴力即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<map>
#define N 200003
#define LL long long 
using namespace std;
map<int,int> mp[1003];
int belong[N],a[N],l[N],r[N],f[N],g[N],n,m;
int gcd(int x,int y){
    int r;
    while (y) {
        r=x%y;
        x=y; y=r;
    }
    return x;
}
int read()
{
    char c=getchar(); int x=0;
    while (c<'0'||c>'9') c=getchar();
    while (c<='9'&&c>='0') x=x*10+c-'0',c=getchar();
    return x;
}
int main()
{
    freopen("gcdxor.in","r",stdin);
    freopen("gcdxor.out","w",stdout);
    scanf("%d",&n);
    for (int i=1;i<=n;i++) a[i]=read();
    int blocksize=ceil(sqrt(n));
    for (int i=1;i<=n;i++){
        belong[i]=(i-1)/blocksize+1;
        if (!l[belong[i]]) l[belong[i]]=i;
        r[belong[i]]=i;
    }
    for (int i=1;i<=blocksize;i++) {
        g[l[i]]=f[l[i]]=a[l[i]];
        if (!mp[i][f[l[i]]]) mp[i][f[l[i]]]=l[i];
        for (int j=l[i]+1;j<=r[i];j++){
         g[j]=gcd(g[j-1],a[j]),f[j]=f[j-1]^a[j];
         if (!mp[i][f[j]])mp[i][f[j]]=j;
        }
    }
    scanf("%d",&m);
    for (int k=1;k<=m;k++){
        char s[10];scanf("%s",s+1);
        if (s[1]=='M') {
            int x,v; x=read(); v=read(); x++;
            int t=belong[x]; a[x]=v; mp[t].clear();
            int L=l[t];
            g[L]=f[L]=a[L];
            if (!mp[t][f[L]]) mp[t][f[L]]=L;
            for (int j=L+1;j<=r[t];j++){
              g[j]=gcd(g[j-1],a[j]),f[j]=f[j-1]^a[j];
              if (!mp[t][f[j]])mp[t][f[j]]=j;
            }
        }
        else {
            LL x; scanf("%lld",&x);
            int lastgcd=0,lastxor=0,ans=0;
            for (int i=1;i<=blocksize;i++)
             if (i!=1&&gcd(lastgcd,g[r[i]])==lastgcd){
                if (x%lastgcd) {
                    lastxor^=f[r[i]];
                    continue;
                }
                LL k=(LL)(x/lastgcd)^lastxor;
                if (k<=(1<<30)&&mp[i][k]) {
                    ans=mp[i][k];
                    break;
                }
                lastxor^=f[r[i]];
             }
             else {
                for (int j=l[i];j<=r[i];j++){
                    lastgcd=gcd(lastgcd,a[j]);
                    lastxor=lastxor^a[j];
                    if ((LL)lastgcd*(LL)lastxor==x){
                        ans=j;
                        break;
                    }
                }
                if (ans) break;
             }
             if (!ans) printf("no\n");
             else printf("%d\n",ans-1);
        }
    }
}