区间最大公约数 - 区间修改、区间查询gcd

给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

1. C l r d，表示把 A[l],A[l+1],…,A[r] 都加上 d。
2. Q l r，表示询问 A[l],A[l+1],…,A[r] 的最大公约数(GCD)。

对于每个询问，输出一个整数表示答案。

输入格式

第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

N≤500000,M≤100000,
1≤A[i]≤1018,
|d|≤1018,
保证数据在计算过程中不会超过 long long 范围。

输入样例：

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

 

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'

using namespace std;

typedef pair<int, int> PII;
typedef long long ll;

const int N = 500010;

int n, m;
ll a[N];
struct Node
{
	int l, r;
	ll sum, gcd;
} tr[N * 4];

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(tr[u], tr[u << 1], tr[u << 1 | 1]);
}

void build(int u, int l, int r)
{
	if(l == r)
	{
		ll b = a[l] - a[l - 1];
		tr[u] = {l, r, b, b};
	}
	else
	{
		tr[u].l = l, tr[u].r = 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, ll c)
{
	if(tr[u]. l == x && tr[u].r == x)
	{
		ll b = tr[u].sum + c;//注意不能写成a[x] - a[x - 1] + c 
		tr[u] = {x, x, b, b};
	}
	else
	{
		int mid = tr[u].l + tr[u].r >> 1;
		if(x <= mid)modify(u << 1, x, c);
		else modify(u << 1 | 1, x, c);
		pushup(u);
	}
}

Node query(int u, int l, int r)
{
	if(tr[u].l >= l && tr[u].r <= r)return tr[u];
	else
	{
		int mid = tr[u].l + tr[u].r >> 1;
		if(l > mid)return query(u << 1 | 1, l, r);
		else if(r <= mid)return query(u << 1, l, r);
		else
		{
			Node left = query(u << 1, l, r);
			Node right = query(u << 1 | 1, l, r);
			Node res;
			pushup(res, left, right);
			return res;
		}
	}
}

int main()
{
	IOS
	cin >> n >> m;
	for(int i = 1; i <= n; i ++)cin >> a[i];
	build(1, 1, n);
	
	char op[2];
	ll l, r, z;
	while(m --)
	{
		cin >> op >> l >> r;
		if(*op == 'Q')
		{
			Node left = query(1, 1, l);
			Node right = {0, 0, 0, 0};
			if(l + 1 <= r)right = query(1, l + 1, r);
			cout << abs(__gcd(left.sum, right.gcd)) << endl;
		}
		else
		{
			cin >> z;
			modify(1, l, z);
			if(r + 1 <= n)modify(1, r + 1, -z);
		}
	}
	
	return 0;
}

 gcd(a1, a2, a3, a4) = |gcd(a1, a2 - a1, a3 - a2, a4 - a3)|

这道题的区间修改是用差分实现的，注意求区间gcd时第一个数不能转化为a1 - ax

线段树对操作的要求很严格，写代码的时候注意不要查询或者修改不存在的点。

void modify(int u, int x, ll c)
{
	if(tr[u]. l == x && tr[u].r == x)
	{
		ll b = tr[u].sum + c;//注意不能写成a[x] - a[x - 1] + c 
		tr[u] = {x, x, b, b};
	}
	else
	{
		int mid = tr[u].l + tr[u].r >> 1;
		if(x <= mid)modify(u << 1, x, c);
		else modify(u << 1 | 1, x, c);
		pushup(u);
	}
}

modify修改某个数时注意不能涉及原数组，如果涉及到二次修改就wa了。

题目：246. 区间最大公约数 - AcWing题库