题意·:给你一个序列,然后有q次操作,每次修改一个数,求有所有子区间不同gcd个个数。
首先预处理出原序列的子区间gcd个数。枚举每个i为左端点,然后以这个为左端点的区间最多有log(n)个,因为a[i]有不超过
log(n)个素因子。然后线段树上二分查找这logn个不同的区间。
然后修改i之会影响含a[i]的子序列,因为i为右端点向左最多有logn个不同的区间,向右也最多有logn个不同的区间,所有最多有logn*logn个区间,依旧线段树上二分查找。
ps:一开始写的是线段树+二分,longn*logn*logn,T了,后来学了一下线段树上二分,确实快好多。
#include <bits/stdc++.h>
using namespace std;
#define ls i*2+1
#define rs i*2+2
#define lson l,mid,ls
#define rson mid+1,r,rs
#define mid (l+r)/2
#define pb push_back
#define root 1,n,0
#define N 200005
#define pa pair<int,int>
#define mk(a,b) make_pair(a,b)
int gcd[N];
int a[N],n,ans;
void build(int l,int r, int i)
{
gcd[i] = a[l];
if(l==r)return ;
build(lson);
build(rson);
gcd[i]=__gcd(gcd[ls],gcd[rs]);
}
int query(int L,int R,int l,int r,int i)
{
//cout<<L<<' '<<R<<endl;
if(l==L&&r==R)return gcd[i];
if(R<=mid)return query(L,R,lson);
if(L>mid)return query(L,R,rson);
return __gcd(query(L,mid,lson),query(mid+1,R,rson));
}
pa queryl(int pos,int v,int l,int r,int i){
if(pos>=r&&(l==r||gcd[i]%v==0))return pa(__gcd(v,gcd[i]),l);
if(pos<=mid)return queryl(pos,v,lson);
pa res=queryl(pos,v,rson);
if(res.first<v)return res;
return queryl(pos,v,lson);
}
pa queryr(int pos,int v,int l,int r,int i){
if(pos<=l&&(l==r||gcd[i]%v==0))return pa(__gcd(v,gcd[i]),r);
if(pos>mid)return queryr(pos,v,rson);
pa res=queryr(pos,v,lson);
if(res.first<v)return res;
return queryr(pos,v,rson);
}
void updata(int pos,int l,int r,int i)
{
gcd[i] = a[pos];
if(l==r) return ;
if(pos<=mid)updata(pos,lson);
else updata(pos,rson);
gcd[i]=__gcd(gcd[ls],gcd[rs]);
}
vector<pa>gl,gr;
void getl(int pos)
{
for(int x=pos,v=a[x];x>=1;){
pa y=queryl(x,v,root);
if(y.first==v)y.second--;
gl.pb(pa(v,x-y.second));
x=y.second;v=y.first;
}
}
void getr(int pos)
{
for(int x=pos,v=a[pos];x<=n;){
pa y=queryr(x,v,root);
if(y.first==v)y.second++;
gr.pb(pa(v,y.second-x));
x=y.second;v=y.first;
}
}
int num[N*5];
void add(int x,int y){
if(!num[x])ans++;
num[x]+=y;
if(!num[x])ans--;
}
void cal(int pos,int f)
{
gl.clear();gr.clear();
getl(pos);getr(pos);
for(pa x:gl)
for(pa y:gr)
add(__gcd(x.first,y.first),f*x.second*y.second);
}
void init()
{
ans=0;
gr.clear();
memset(num,0,sizeof(num));
for(int i=1;i<=n;i++) getr(i);
for(pa x:gr)add(x.first,x.second);
}
int main()
{
int T,cas=1,q;
cin>>T;
while(T--){
printf("Case #%d:\n",cas++);
scanf("%d %d",&n,&q);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(root);
init();
while(q--){
int pos,x;
scanf("%d %d",&pos,&x);
cal(pos,-1);
a[pos]=x;
updata(pos,root);
cal(pos,1);
printf("%d\n",ans);
}
}
return 0;
}