246. 区间最大公约数(线段树,gcd,《算法竞赛进阶指南》)

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

给定一个长度为 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

解析: 

gcd部分:

一般而言要证明 ,要证明 a=b,可以转化为证明 a>=b,b>=a.

证明 gcd(a1,a2,...,an)<=gcd (a1,a2-a1,a3-a2,...,an-an-1);

令 d = gcd(a1,a2,…,an),

则 a1%d=0,a2%d=0,…,an%d=0

则 a1%d=0,(a2-a1)%d=0,(a3-a2)%d=0,…,(an-an-1)%d=0

所以 gcd(a1,a2,...,an)<=gcd(a1,a2-a1,a3-a2,...,an-an-1);(子集包含关系)

类似的,可以证明>= 关系。

详细解释参考蓝书

#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<math.h>
#include<map>
#include<sstream>
#include<deque>
#include<unordered_map>
#include<unordered_set>
#include<bitset>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
typedef pair<double, double> PDD;
const int INF = 0x3f3f3f3f;
const LL Mod = 1e9;
const int N = 5e5 + 10, M = 250 + 10, P = 110;
int n, m;
LL w[N];
struct Node {
	int l, r;
	LL sum, d;
}tr[N*4];

LL gcd(LL a, LL b) {
	if (b == 0)return a;
	return gcd(b, a % b);
}

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) {
		LL b = w[l] - w[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);
	}
}

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 (r <= mid)return query(u << 1, l, r);
		else if (l > mid)return query(u << 1 | 1, l, r);
		else {
			Node ret;
			Node left = query(u << 1, l, r);
			Node right = query(u << 1 | 1, l, r);
			pushup(ret, left, right);
			return ret;
		}
	}
}



void modify(int u, int x, LL c) {
	if (tr[u].l == x && tr[u].r == x) {
		LL d = tr[u].sum + c;
		tr[u] = { x,x,d,d };
	}
	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);
	}
}


int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		scanf("%lld", &w[i]);
	}
	build(1, 1, n);
	char op[2];
	int l, r;
	LL d;
	while (m--) {
		scanf("%s%d%d", &op, &l, &r);
		if (op[0] == 'Q') {
			auto left = query(1, 1, l);
			//printf("+____________%d\n", left.sum);
			Node right({ 0, 0, 0, 0 });
			if (l + 1 <= r) right = query(1, l + 1, r);
			printf("%lld\n", abs(gcd(left.sum, right.d)));
		}
		else {
			scanf("%lld", &d);
			modify(1, l, d);
			if(r+1<=n)modify(1, r + 1, -d);
		}
	}
	return 0;
}

  • 5
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值