1. 程式人生 > >[HEOI2015]公約數數列

[HEOI2015]公約數數列

turn 前綴 onclick pla post blank lan 輸入 lose

公約數數列

【問題描述】

設計一個數據結構. 給定一個正整數數列 a_0, a_1, ...,a_{n - 1},你需要支持以下兩種操作:

1. MODIFY id x: 將 aid 修改為 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表示最大公約數。

【輸入格式】

從gcdxor.in中讀入。

輸入數據的第一行包含一個正整數 n.

接下來一行包含 n 個正整數 a0,a1,...,an1.

之後一行包含一個正整數 q,表示詢問的個數。

之後 q 行,每行包含一個詢問。格式如題目中所述。

【輸出格式】

輸出到gcdxor.out中

對於每個 QUERY 詢問,在單獨的一行中輸出結果。如果不存在這樣的 p,輸出 no.

【樣例輸入】

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

【樣例輸出】

6

0

no

2

8

8

【數據規模與約定】

對於 30% 的數據,n <= 10000,q <= 1000.

對於 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.

這道題應該還是比較容易往分塊那裏去想的。

然而我還是看了題解。

對於代碼中各變量的含義:len塊大小,cnt塊個數,pos當前位置屬於哪個塊,l第i個塊的左端點,r右端點,xx第i個塊所有元素異或和,g第i個塊所有元素gcd。

首先我們需要知道一個性質,就是前綴的gcd是單調不增的。那麽對於第i個塊,令前i-1個塊的gcd為x,前i個塊gcd為y,如果x==y,那麽意味著i這個塊內所有元素的前綴gcd都為x,因為前綴gcd是單調不增的。那麽我們如果知道前i-1個塊的異或和是多少,就可以判斷當前塊i中是否存在一個位置滿足題意。記前i-1個塊的異或和為lsxor(枚舉塊時記錄即可),要查詢的值為H,那麽如果存在一個位置j符合題意,記u為j到其所在塊左端點的異或和,則有x*(lsxor^u)=H,u=(H/x)^lsxor。右面的數字是已知的,所以我們可以用一個map來判斷是否存在這樣一個j。復雜度log

如果x!=y,暴力枚舉查詢即可。因為每一次x!=y,y至少變為x的一半,也就是最多出現log次這種情況。所以復雜度是sqrt(n)log^2。

對於修改操作,我們暴力修改並重新更新map,xx,g等數組即可,復雜度sqrt(n)log。

所以整體復雜度大概是nsqrt(n)log^2的,雖然我不知道為什麽能過...講道理用map會tle,但cogs氧氣足...

技術分享圖片
 1 #include<bits/stdc++.h>
 2 #define ll long long
 3 using namespace std;
 4 const int inf=100005;
 5 int n,q,a[inf];
 6 int len,cnt,pos[inf],l[inf/100],r[inf/100],xx[inf/100],g[inf/100];
 7 map<int,int>S[inf/100];
 8 int gcd(int x,int y){
 9     return !y?x:gcd(y,x%y);
10 }
11 void change(){
12     int id,x;
13     scanf("%d%d",&id,&x);
14     id++;
15     int now=pos[id];
16     S[now].clear();
17     a[id]=x;
18     int xo=0,gc=a[l[now]];
19     for(int i=l[now];i<=r[now];i++){
20         xo^=a[i];
21         gc=gcd(gc,a[i]);
22         if(S[now].find(xo)==S[now].end())S[now][xo]=i;
23     }
24     g[now]=gc;
25     xx[now]=xo;
26 }
27 int work(){
28     ll x;
29     scanf("%lld",&x);
30     int lsgcd=a[1],lsxor=0;
31     for(int i=1;i<=r[1];i++){
32         lsxor^=a[i];
33         lsgcd=gcd(lsgcd,a[i]);
34         if(1ll*lsxor*lsgcd==x)return i;
35     }
36     for(int i=2;i<=cnt;i++){
37         if(gcd(g[i],lsgcd)==lsgcd){
38             if(x%lsgcd==0){
39                 int u=(1ll*x/lsgcd)^lsxor;
40                 if(S[i].find(u)!=S[i].end())return S[i][u];
41             }
42             lsgcd=gcd(lsgcd,g[i]);
43             lsxor^=xx[i];
44         }
45         else {
46             for(int j=l[i];j<=r[i];j++){
47                 lsxor^=a[j];
48                 lsgcd=gcd(lsgcd,a[j]);
49                 if(1ll*lsxor*lsgcd==x)return j;
50             }
51         }
52     }
53     return 0;
54 }
55 int main()
56 {
57     scanf("%d",&n);
58     for(int i=1;i<=n;i++)scanf("%d",&a[i]);
59     len=sqrt(n);
60     for(int i=1;i<=n;i++){
61         cnt=pos[i]=(i-1)/len+1;
62         if(pos[i]!=pos[i-1])l[cnt]=i;
63         r[cnt]=i;
64     }
65     for(int i=1;i<=cnt;i++){
66         int xo=0,gc=a[l[i]];
67         for(int j=l[i];j<=r[i];j++){
68             xo^=a[j];
69             gc=gcd(gc,a[j]);
70             if(S[i].find(xo)==S[i].end())S[i][xo]=j;
71         }
72         g[i]=gc;
73         xx[i]=xo;
74     }
75     scanf("%d",&q);
76     for(int i=1;i<=q;i++){
77         char s[20];
78         scanf("%s",s);
79         if(s[0]==M)change();
80         else {
81             int ans=work();
82             if(ans)printf("%d\n",ans-1);
83             else printf("no\n");
84         }
85     }
86     return 0;
87 }
View Code

[HEOI2015]公約數數列