题意:
思路:
题目中的要求是 区间修改:整个区间都加上一个数 和 区间查询:查询一段区间中的最大公约数。
对于区间修改,我们可以将原数组变换为差分数组,因此可以将区间修改转化成了线段树基本操作之一的单点修改。
对于原数组求最大公约数,我们根据辗转相除法可以得知:gcd(a[1], a[2], a[3], ..., a[i-1], a[i]) = gcd(b[1] = a[1], b[2] = a[2]-a[1], ..., b[i] = a[i]-a[i-1])
(b
数组表示a
数组的差分数组),
区间[l, r]
的最大公约数查询结果为:gcd(a[l], gcd(b[l+1], b[l+2], ..., b[r])
(a[l]
可以通过差分数组b
1~l
求前缀和sum
而得)
即:原数组所有元素的最大公约数 等于 其差分数组最大公约数。
显然我们要在线段树每个节点内部存储额外两个信息:sum
区间和、Gcd
最大公约数。
其中父节点u.sum = L_son.sum + R_son.sum
,u.Gcd = gcd(L_son.Gcd, R_son.Gcd)
至此,我们就将本题中要求的两个操作完全转化成了线段树的基本操作。
注意点:
-
①差分操作是
a[l]+d,a[r+1]-d;
这个时候就可能出现r+1
越界的情况。这非常重要,在main
主函数的查询和修改时都要及时地特判掉。 -
②在数值加减的过程中可能会产生负数,而约定gcd是没有负数的,所以需要用这个式子来搞定负数:
gcd(a,b)=gcd(a,−b)
具体来说,就是在每次查询或者更新的时候,如果遇到了负数,就把它取相反数。
注意只能对结果res取反而不能直接把线段树的负数叶子节点取相反数。因为直接把叶子取反会对今后的加减操作造成影响(因为虽然gcd(a,b)=gcd(a,−b)
但是(a+1,b)
和(−a+1,b)
不一定相等。) -
③当然,那些
pushup
函数等其他地方的基本变化还是要有的,具体可以见代码~
代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int N = 5e5+10;
int a[N];
int n, m;
struct node
{
int l, r;
int sum, Gcd;
} t[N<<2];
void pushup(node &u, node &l, node &r)
{
u.sum = l.sum + r.sum;
u.Gcd = __gcd(l.Gcd, r.Gcd);
}
void pushup(int u){
pushup(t[u], t[u<<1], t[u<<1|1]);
}
void build(int u, int l, int r)
{
t[u] = {l, r};
if(l==r) {t[u].sum = a[l]-a[l-1], t[u].Gcd = a[l]-a[l-1]; return ;}
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(t[u].l==t[u].r) {t[u] = {x, x, t[u].sum+v, t[u].Gcd+v}; return ;}
int mid = t[u].l + t[u].r >> 1;
if(x<=mid) modify(u<<1, x, v);
else modify(u<<1|1, x, v);
pushup(u);
}
node ask(int u, int l, int r)
{
if(l<=t[u].l&&r>=t[u].r) return t[u];
int mid = t[u].l + t[u].r >>1;
if(r<=mid) return ask(u<<1, l, r);
else if(l>=mid+1) return ask(u<<1|1, l, r);
else
{
auto left = ask(u<<1, l, r);
auto right = ask(u<<1|1, l, r);
node res;
pushup(res, left, right);
return res;
}
}
signed main()
{
scanf("%lld%lld", &n, &m);
for( int i=1; i<=n; ++i ) scanf("%lld", &a[i]);
build(1, 1, n);
while(m--)
{
char op[2];
int l, r, d;
scanf("%s%lld%lld", op, &l, &r);
if(*op=='Q')
{
node left = ask(1, 1, l), right = {0, 0, 0, 0};
if(r + 1 <= n) right = ask(1, l + 1, r);//一定要及时特判,如这个条件不满足就默认为全0
int res = __gcd(left.sum, right.Gcd);
printf("%lld\n", abs(res));//由于可能存在负数,我们规定gcd为正数,因此要取绝对值
}
else
{
scanf("%lld", &d);
modify(1, l, d);
if(r + 1 <= n) modify(1, r + 1, -d);//一定要及时特判
}
}
return 0;
}
新发现的一条路,来自另一位博主: