acwing 246 区间最大公约数
题目链接:传送门
输入样例:
5 5
1 3 5 7 9
Q 1 5
C 1 5 1
Q 1 5
C 3 3 6
Q 2 4
输出样例:
1
2
4
这道题有很多坑点,第一,注意开long long ,防止爆int 第二,对于边界要进行限制,否则会越界
思路:
一个数学知识:gcd(a1 , a2 , a3 … an) = gcd(a1 , a2 - a1 , … , an - an-1)
由此,我们可以进行优化, 对于区间求[l,r]的最大公约数,可以拆分成 aL 和 a[L+1… R]两部分的最大公约数进行取gcd
对于后者a2 - a1 , … , an - an-1,利用线段树维护差分数组,
对于前者aL, 求aL,就是求区间 1到L 差分上的的前缀和,这部分可以用树状数组维护,也可以用线段树维护。
代码:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
#define int long long
const int N = 500010;
struct node{
int l,r;
int sum;
int d;
}tr[N<<2];
int w[N];
int gcd(int a,int b){
return b?gcd(b,a%b) : a;
}
void pushup(node &u,node &l,node &r){
u.sum = l.sum + r.sum;
u.d = gcd(l.d,r.d);
}
void pushup(int u){
pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}
void build(int u,int l,int r){
if(l == r){
int b = w[r]-w[r-1];
tr[u] = {r,r,b,b};
}else{
tr[u] ={l,r};
int mid = l+r >> 1;
build(u<<1,l,mid);
build(u<<1|1,mid+1,r);
pushup(u);
}
}
void modify(int u,int x,int v){
if(tr[u].l == x && tr[u].r == x){
int b = tr[u].sum + v;
tr[u] = {x,x,b,b};
}else{
int mid = tr[u].l + tr[u].r >> 1;
if(x<=mid) modify(u<<1,x,v);
else modify(u<<1|1,x,v);
pushup(u);
}
}
node query(int u,int l,int r){
if(tr[u].l>=l && tr[u].r <= r) return tr[u];
int mid = tr[u].l + tr[u].r >> 1;
if(r<=mid) return query(u<<1,l,r);
else if(l>mid) return query(u<<1|1,l,r);
else{
auto left = query(u<<1,l,r);
auto right = query(u<<1|1,l,r);
node res;
pushup(res,left,right);
return res;
}
}
signed main()
{
int n,m;
cin>>n>>m;
for (int i = 1; i <= n; i ++ ) cin>>w[i];
build(1,1,n);
while (m -- ){
char op[2]; int l,r;
cin>>op>>l>>r;
if(*op =='Q'){
int left = 0;
left = query(1,1,l).sum;
int right = 0;
if(l+1<=n) right = query(1,l+1,r).d;
cout<< abs(gcd(left,right))<<endl;
}
else{
int x; cin>>x;
modify(1,l,x);
if(r+1<=n) modify(1,r+1,-x);
}
}
}