BZOJ4028 [HEOI2015]公约数数列

Description

设计一个数据结构. 给定一个正整数数列 a_0, a_1, ..., a_{n - 1}，你需要支持以下两种操作：

1. MODIFY id x: 将 a_{id} 修改为 x.

2. QUERY x: 求最小的整数 p (0 <= p < n)，使得 gcd(a_0, a_1, ..., a_p) * XOR(a_0, a_1, ..., a_p) = x. 其中 XOR(a_0, a_1, ..., a_p) 代表 a_0, a_1, ..., a_p 的异或和，gcd表示最大公约数。

Input

 输入数据的第一行包含一个正整数 n.

接下来一行包含 n 个正整数 a_0, a_1, ..., a_{n - 1}.

之后一行包含一个正整数 q，表示询问的个数。

之后 q 行，每行包含一个询问。格式如题目中所述。

Output

对于每个 QUERY 询问，在单独的一行中输出结果。如果不存在这样的 p，输出 no.

Sample Input

10
1353600 5821200 10752000 1670400 3729600 6844320 12544000 117600 59400 640
10
MODIFY 7 20321280
QUERY 162343680
QUERY 1832232960000
MODIFY 0 92160
QUERY 1234567
QUERY 3989856000
QUERY 833018560
MODIFY 3 8600
MODIFY 5 5306112
QUERY 148900352

Sample Output

6
0
no
2
8
8

HINT

 对于 100% 的数据，n <= 100000，q <= 10000，a_i <= 10^9 (0 <= i < n)，QUERY x 中的 x <= 10^18，MODIFY id x 中的 0 <= id < n，1 <= x <= 10^9.

 

正解：分块

解题报告：　　

　　看到数据范围我这种蒟蒻居然第一印象是分块，我真是太弱了。结果一看正解果然是分块，这就非常的优秀了。

　　维护每个块内的gcd、前缀异或和，修改的时候暴力修改块内即可。

　　考虑查询。因为我们要求一个前缀异或和×前缀gcd等于给定值y的最小下标，那么我们考虑做当前块的时候，如果当前gcd与之前所有的gcd（设为lastgcd）的gcd相等，由于前缀gcd一定是不增的。所以相等就意味着这个块的所有前缀gcd和lastgcd的gcd都是lastgcd，当前异或前缀和设为lastxor，存在在第i块中的前j个异或前缀和为X[i][j],则

　　lastgcd × （X[i][j]^lastxor）=y

　　则X[i][j]=(y/lastgcd)^lastxor，右边都是已知的

　　只需在当前块中查找有没有这种元素。

　　原来用的是map，居然TLE了，然后改成数组每次排序就可以了。调试了很久，写错了很多地方。。。

 

　　　　

//It is made by jump~
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <ctime>
#include <vector>
#include <queue>
#include <map>
#include <set>
#ifdef WIN32   
#define OT "%I64d"
#else
#define OT "%lld"
#endif
using namespace std;
typedef long long LL;
const int MAXN = 100011;
int n,m,kuai,tot;
int belong[MAXN],L[MAXN],R[MAXN];
char ch[10];
LL G[401],X[401][401];//gcd、xor
LL a[MAXN];

struct node{
    LL val;
    int x;
}b[MAXN];

inline int getint()
{
    int w=0,q=0;
    char c=getchar();
    while((c<'0' || c>'9') && c!='-') c=getchar();
    if (c=='-')  q=1, c=getchar();
    while (c>='0' && c<='9') w=w*10+c-'0', c=getchar();
    return q ? -w : w;
}

inline LL getlong()
{
    LL w=0,q=0;
    char c=getchar();
    while((c<'0' || c>'9') && c!='-') c=getchar();
    if (c=='-')  q=1, c=getchar();
    while (c>='0' && c<='9') w=w*10+c-'0', c=getchar();
    return q ? -w : w;
}

inline LL gcd(LL x,LL y){ if(y==0) return x; return gcd(y,x%y); }
inline bool cmp(node q,node qq){ if(q.val==qq.val) return q.x<qq.x; return q.val<qq.val; }

inline void work(){
    n=getint(); for(int i=1;i<=n;i++) a[i]=getint();
    kuai=sqrt(n); tot=n/kuai; if(n%kuai) tot++;
    for(int i=1;i<=n;i++) {
    belong[i]=(i-1)/kuai+1;
    if(belong[i]!=belong[i-1]) L[belong[i]]=i;
    R[belong[i]]=i;
    }
    int now;
    for(int i=1;i<=n;i++)  {
    now=(i-1)%kuai+1; if(now==1) { G[belong[i]]=X[belong[i]][1]=a[i]; continue; }
    G[belong[i]]=gcd(G[belong[i]],a[i]);
    X[belong[i]][now]=X[belong[i]][now-1]^a[i];
    }
    for(int i=1;i<=tot;i++) for(int j=1;j<=kuai;j++) b[(i-1)*kuai+j].val=X[i][j],b[(i-1)*kuai+j].x=(i-1)*kuai+j;
    for(int i=1;i<=tot;i++) sort(b+L[i],b+R[i]+1,cmp);
    m=getint(); int x; LL y;
    LL lastgcd,xr,find; int l,r,mid,wei;
    while(m--) {
    scanf("%s",ch);
    if(ch[0]=='M') {
        x=getint()+1; y=getlong();
        //mp[belong[x]].clear(); 
        a[x]=y;   X[belong[x]][1]=G[belong[x]]=a[L[belong[x]]];
        for(int i=L[belong[x]]+1;i<=R[belong[x]];i++) G[belong[x]]=gcd(G[belong[x]],a[i]),X[belong[x]][(i-1)%kuai+1]=X[belong[x]][(i-1)%kuai]^a[i];
        for(int i=L[belong[x]];i<=R[belong[x]];i++) b[i].val=X[belong[x]][(i-1)%kuai+1],b[i].x=i;
        sort(b+L[belong[x]],b+R[belong[x]]+1,cmp);
    }else{
        y=getlong(); lastgcd=0; xr=0;
        bool ok=false;
        for(int i=1;i<=tot;i++) {
        if(i==1 || gcd(G[i],lastgcd)!=lastgcd) {//暴力做
            for(int j=1;j<=kuai;j++) {
            lastgcd=gcd(lastgcd,a[j+(i-1)*kuai]);
            xr^=a[j+(i-1)*kuai];
            if(xr*lastgcd==y) { printf("%d\n",(i-1)*kuai+j-1); ok=true; break; }
            }
            if(ok) break;
        }
        else{//可以利用性质直接查找
            LL search=y/lastgcd; 
            if(y%lastgcd==0)  {
            find=search^xr; 
            l=L[i],r=R[i]; wei=0;
            while(l<=r) { mid=(l+r)/2; if(b[mid].val>=find) wei=mid,r=mid-1; else l=mid+1; }
            if(b[wei].val==find)  {  printf("%d\n",b[wei].x-1); ok=true; }            
            }
            xr^=X[i][kuai]; lastgcd=gcd(lastgcd,G[i]);
        }
        if(ok)  break;         
        }
        if(!ok) printf("no\n");
    }
    }
}

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